比赛 2009noip模拟试卷 评测结果 AAAAAAAAAA
题目名称 火车站饭店 最终得分 100
用户昵称 Smile 运行时间 0.438 s
代码语言 C++ 内存使用 2.98 MiB
提交时间 2016-10-09 15:09:07
显示代码纯文本
#include<iostream>
#include<cstdio>
#include<vector>
#include<cstring>
using namespace std;

const int maxn=100010;

int v[maxn];
vector<int> G[maxn];
int Y[maxn], B[maxn], vis[maxn];

void dp(int root, int fa)
{
	for(int i=0; i<G[root].size(); i++) {
		int x=G[root][i];
		if(x!=fa) {
			if(!vis[x]) dp(x, root);// 这句话很重要, 看的一个题解上没写, 
			vis[x]=1;               // 还好我机智自己加上啦 -_-
			Y[root]+=B[x];
			B[root]+=max(Y[x], B[x]);
		}
	}
	Y[root]+=v[root];
}
// 哈哈哈, 树形动规也不过如此!; 就是分两种状态搜――搜――搜―― 
int main()
{
	freopen("profitz.in", "r", stdin);
	freopen("profitz.out", "w", stdout);
	int n;
	scanf("%d", &n);
	for(int i=1; i<=n; i++) scanf("%d", &v[i]);
	for(int i=1; i<n; i++) {
		int x, y;
		scanf("%d%d", &x, &y);
		G[x].push_back(y);
		G[y].push_back(x);
	}
	dp(1, 0);
	cout<<max(Y[1], B[1]);
	return 0;
}