比赛 20151026 评测结果 AAAAAAAAAA
题目名称 火车站饭店 最终得分 100
用户昵称 Satoshi 运行时间 0.419 s
代码语言 C++ 内存使用 2.70 MiB
提交时间 2015-10-26 21:59:32
显示代码纯文本
#include <fstream>
#include <vector>
#include <algorithm>
using namespace std;
typedef long long ll;
ifstream in("profitz.in");
ofstream out("profitz.out");
int n;
int w[100001]={0};
bool l[100001]={0};
vector<int> G[100001];
int ans=0,f[100001][2]={0};
void read()
{
	int i,j,a,b;
	in>>n;
	for(i=1;i<=n;i++)
	{
		in>>w[i];
		//out<<w[i]<<' ';
	}
	for(i=1;i<n;i++)
	{
		in>>a>>b;
		G[a].push_back(b);
		G[b].push_back(a);
	}
}
ll DFS(int u)
{
	int i,j,v;
	l[u]=1;
	//out<<u<<' ';
	for(i=0;i<G[u].size();i++)
	{
		v=G[u][i];
		if(!l[v])
		{
			DFS(v);
		    f[u][1]+=f[v][0];
		    f[u][0]+=max(f[v][0],f[v][1]);
		}
	}
}
void work()
{
	int i,j;
	for(i=1;i<=n;i++)f[i][1]=w[i];
	DFS(1);
	ans=max(f[1][0],f[1][1]);
	out<<ans<<endl;
}
int main()
{
	read();
	work();
	return 0;
}