比赛 2019级快乐小组模拟赛19.9.19 评测结果 AAAAAAAAAA
题目名称 没有上司的舞会 最终得分 100
用户昵称 梦那边的美好ET 运行时间 0.033 s
代码语言 C++ 内存使用 13.87 MiB
提交时间 2019-09-19 12:35:29
显示代码纯文本
#include<iostream>
#include<cmath>
#include<cstdlib>
#include<cstdio>
#include<algorithm>
#include<cstring>
#include<vector>
#define maxn 6010
using namespace std;
vector<int>b[maxn];
int n,a1,a2,val[maxn],f[maxn][2];
void dfs(int p,int fa){
	f[p][1]=val[p];
	for(int i=0;i<b[p].size();i++){
		int x=b[p][i];if(x==fa)continue;
		dfs(x,p);
		f[p][0]+=max(f[x][0],f[x][1]);
		f[p][1]+=f[x][0];
	}
	return;
}
int main(){
	freopen("partyy.in","r",stdin);
	freopen("partyy.out","w",stdout);
	scanf("%d",&n);
	for(int i=1;i<=n;i++)scanf("%d",&val[i]);
	for(int i=1;i<=n;i++){
		scanf("%d%d",&a1,&a2);
		b[a1].push_back(a2);
		b[a2].push_back(a1);
	}
	dfs(1,0);
	printf("%d\n",max(f[1][0],f[1][1]));
	return 0;
}