蓝桥杯2015届C++B组省赛真题 生命之树

题目：　　

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

他给每棵树的每个节点（叶子也称为一个节点）上，都标了一个整数，代表这个点的和谐值。
上帝要在这棵树内选出一个非空节点集S，使得对于S中的任意两个点a,b，都存在一个点列 {a, v1, v2, ..., vk, b} 使得这个点列中的每个点都是S里面的元素，且序列中相邻两个点间有一条边相连。

在这个前提下，上帝要使得S中的点所对应的整数的和尽量大。
这个最大的和就是上帝给生命之树的评分。

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

「输入格式」
第一行一个整数 n 表示这棵树有 n 个节点。

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

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

　　「样例输入」
　　5
　　1 -2 -3 4 5
　　4 2
　　3 1
　　1 2
　　2 5

　　「样例输出」
　　8

　　「数据范围」
　　对于 30% 的数据，n <= 10
　　对于 100% 的数据，0 < n <= 10^5, 每个节点的评分的绝对值不超过 10^6 。

分析：

这道题难点在于它是一颗无根树，如果把他化成有根树来做会不会简单一点

那到底以谁作为根节点呢？

 那我就以1号节点作为根节点

题型：树型dp问题

dp[u][0]表示不选取u节点后u子树能得到的最大值

dp[u][1]表示选取u节点后的u子树能得到的最大值

w[u]表示权重

dp[u][0] = max(dp[v][0]，dp[v][1]) ，v是u的儿子

dp[u][1] = w[u] + ，v是u的儿子

问题1：这里为什么要和0比较？

因为如果dp[v]是正数，无论w[u]是负数还是正数,加上后都会使值增大，但如果dp[v]是负数，那它只会拖累我，果断舍弃

问题2：为什么这里只写dp[v][1]?

因为S点集中的任意两个点必须是连通的，如果不选v节点，u，v两节点就会不连通

代码：

#include<iostream>
#include<cmath>
#include<vector>
using namespace std;
int w[100001];
vector<int> G[100001];//邻接表 
long long dp[100001][2];
long long ans;

void dfs(int u,int fa){
	dp[u][0] = 0;
	dp[u][1] = w[u];
	for(int i=0;i<G[u].size();i++){
		int v = G[u][i];//u节点相邻的节点 
		if(v != fa){//父节点也与u节点相邻，但不能重复遍历 
			dfs(v,u);//dfs后才能获取儿子的信息，dp[v]才能知晓 
			dp[u][0] = max(dp[v][0],dp[v][1]);
			if(dp[v][1] > 0){
				dp[u][1] += dp[v][1];
			}	 
		}
	}
	long long Max = max(dp[u][0],dp[u][1]);
	ans =  max(Max,ans);//Max有可能是负数，但和谐值要>=0
}
int main(){
	int n;
	cin >> n;
	for(int i=1;i<=n;i++){
		cin >> w[i]; 
	}
	int a,b;
	for(int i=1;i < n;i++){
		cin >> a >> b;
		G[a].push_back(b);
		G[b].push_back(a);
	}
	dfs(1,1);
	cout << ans;
	return 0; 
}