比赛 20111109 评测结果 WWWWWWWWWW
题目名称 火车站饭店 最终得分 0
用户昵称 wangmengyuan 运行时间 0.000 s
代码语言 C++ 内存使用 0.00 MiB
提交时间 2011-11-09 11:16:32
显示代码纯文本
#include<iostream>
#include<cstdio>
#include<cstdlib>
#define max(a,b) ((a)>=(b)?(a):(b))
using namespace std;
int w[100001],n,m;
int q[100001][2],a[100001][51];
int ji[100001]={0};
bool used[100001]={0};
void dp(int x);
int main()
{
	freopen ("profitz.in","r",stdin);
	freopen ("profitz.out","w",stdout);
	scanf("%d",&n);
	for (int o=1;o<=n;o++)
	{
		scanf("%d",&w[o]);
	}
	for (int o=1;o<=n-1;o++)
	{
		int c,d;
		scanf("%d%d",&c,&d);
		ji[c]++;
		ji[d]++;
		a[c][ji[c]]=d;
		a[d][ji[d]]=c;
	}
	dp(1);
	if (q[1][0]>q[1][1])
	{
		cout<<q[1][0];
	}
	else
	{
		cout<<q[1][1];
	}
	return 0;
}
void dp(int x)
{
	used[x]=1;
	for (int i=1;i<=ji[x];i++)
	{
 		if (!used[a[x][i]])
		{
			dp(a[x][i]);
			q[x][0]+=max(q[a[x][i]][0],q[a[x][i]][1]);
			q[x][1]+=max(q[x][1],w[x]+q[a[x][i]][0]);
		}
	}
	used[x]=0;
}