比赛 20151026 评测结果 C
题目名称 火车站饭店 最终得分 0
用户昵称 玉带林中挂 运行时间 0.000 s
代码语言 C++ 内存使用 0.00 MiB
提交时间 2017-10-16 21:56:41
显示代码纯文本
#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>
#include <vector>
using namespace std;
const int e=100020;
int f[e]={0},g[e]={0},a[e]={0},link[e]={0};
int n,t=0;
struct qq
{
	int y,next;
}ee[2*e];
void insert(int startt,int endd)
{
	ee[++t].y=endd;ee[t].next=link[startt];
	link[startt]=t;
}
void dfs(int root,int father)
{
	int tempf=0,tempg=0;
	for(int i=link[root];i;i=ee[i].next)
	{
		if(ee[i].y!=father)
		{
			dfs(ee[i].y,root);
			f[root]+=g[ee[i].y];
			g[root]+=max(f[ee[i].y],g[ee[i].y]);
		}	
	}
	f[root]+=a[root];
}
int main()
{
	freopen("profitz.in","r",stdin);freopen("profitz.out","w",stdout);
	cin>>n;
	for(int i=1;i<=n;i++)
		scanf("%d",&a[i]);
	for(int i=1;i<n;i++)
	{
		int xx,yy;
		scanf("%d%d",&xx,&yy);
		insert(xx,yy);
		insert(yy,xx);
	}
	dfs(1,1);
	cout<<max(f[1],g[1]);
	fclose(stdin);fclose(stdout);
	return 0;
}