记录编号 461792 评测结果 AAAAAAAAAA
题目名称 火车站饭店 最终得分 100
用户昵称 Gravatar胖周zzf 是否通过 通过
代码语言 C++ 运行时间 0.434 s
提交时间 2017-10-20 18:45:30 内存使用 2.98 MiB
显示代码纯文本
#include<cstdio>
#include<vector>
#include<algorithm>
#include<cstring>
using namespace std;
struct{
	int c,f;
	vector<int>next;
}x[100010];
int d[100010][2]={0},K;
void TREEDP(int rt,int frt){
	d[rt][1]=x[rt].c;d[rt][0]=0;
	for(int i=0;i<x[rt].next.size();i++){
		if(x[rt].next[i]!=frt&&d[x[rt].next[i]][0]==-1){
			TREEDP(x[rt].next[i],rt);
		}
	}
	for(int i=0;i<x[rt].next.size();i++){
		if(x[rt].next[i]!=frt){
			d[rt][1]+=d[x[rt].next[i]][0];
			d[rt][0]+=max(d[x[rt].next[i]][0],d[x[rt].next[i]][1]);
		}
	}
}
int main(){
	freopen("profitz.in","r",stdin);
	freopen("profitz.out","w",stdout);
	memset(d,-1,sizeof(d));
	int n;scanf("%d",&n);
	for(register int i=1;i<=n;i++)
		scanf("%d",&x[i].c);
	for(register int i=1,a,b;i<n;i++){
		scanf("%d%d",&a,&b);
		x[a].next.push_back(b);
		x[b].next.push_back(a);
	}
	TREEDP(1,0);
	printf("%d",max(d[1][0],d[1][1]));
	return 0;
}