[蓝桥杯 2015 省 B] 生命之树（树形DP）

题目描述

在 X 森林里，上帝创建了生命之树。

他给每棵树的每个节点（叶子也称为一个节点）上，都标了一个整数，代表这个点的和谐值。

上帝要在这棵树内选出一个节点集合 $S$（允许为空集），使得对于 $S$ 中的任意两个点 $a,b$，都存在一个点列 $a,v_1,v_2,\cdots ,v_k,b$ 使得这个点列中的每个点都是 $S$ 里面的元素，且序列中相邻两个点间有一条边相连。

在这个前提下，上帝要使得 $S$ 中的点所对应的整数的和尽量大。

这个最大的和就是上帝给生命之树的评分。

经过 atm 的努力，他已经知道了上帝给每棵树上每个节点上的整数。但是由于 atm 不擅长计算，他不知道怎样有效的求评分。他需要你为他写一个程序来计算一棵树的分数。

输入格式

第一行一个整数 $n$ 表示这棵树有 $n$ 个节点。

第二行 $n$ 个整数，依次表示每个节点的评分。

接下来 $n-1$ 行，每行 $2$ 个整数 $u,v$，表示存在一条 $u$ 到 $v$ 的边。由于这是一棵树，所以是不存在环的。

输出格式

输出一行一个数，表示上帝给这棵树的分数。

样例 #1

样例输入 #1

5
1 -2 -3 4 5
4 2
3 1
1 2
2 5

样例输出 #1

8

提示

对于 $30\%$ 的数据，$n\le 10$。

对于 $100\%$ 的数据，$0<n\le 10^5,$ 每个节点的评分的绝对值不超过 $10^6$。

时限 3 秒, 256M。

蓝桥杯 2015 省赛 B 组 J 题。

解析

树形DP
由于答案唯一，所以任意一点为根节点即可，不妨选择 $1$ 为根节点

代码

#include<bits/stdc++.h>
using namespace std;
#define int long long
const int N=2e5+5;
int n,m,a[N],dp[N]; 
vector<int>e[N];
void dfs(int u,int fa){
	dp[u]=a[u];
	for(int i=0;i<e[u].size();i++){
		int x=e[u][i];
		if(x!=fa){
			dfs(x,u);
			if(dp[x]>0) dp[u]+=dp[x];
		}
	}
}
void solve(){
	scanf("%lld",&n);
	for(int i=1;i<=n;i++){
		scanf("%lld",&a[i]);
	}
	for(int i=1;i<n;i++){
		int x,y;
		scanf("%lld%lld",&x,&y);
		e[x].push_back(y);
		e[y].push_back(x);
	}
	dfs(1,-1);
	int ans=0;
	for(int i=1;i<=n;i++) ans=max(ans,dp[i]);
	cout<<ans;
}
signed main(){
	int t=1;
//	scanf("%d",&t);
	while(t--) solve();
	return 0;
}