1262: [蓝桥杯2015初赛]生命之树

题目描述
在X森林里，上帝创建了生命之树。
他给每棵树的每个节点(叶子也称为一个节点)上，都标了一个整数，代表这个点的和谐值。
上帝要在这棵树内选出一个非空节点集S，使得对于S中的任意两个点a,b，都存在一个点列 {a, v1, v2, …, vk, b}
使得这个点列中的每个点都是S里面的元素，且序列中相邻两个点间有一条边相连。
在这个前提下，上帝要使得S中的点所对应的整数的和尽量大。
这个最大的和就是上帝给生命之树的评分。
经过atm的努力，他已经知道了上帝给每棵树上每个节点上的整数。
但是由于 atm 不擅长计算，他不知道怎样有效的求评分。
他需要你为他写一个程序来计算一棵树的分数。

输入
第一行一个整数 n 表示这棵树有 n 个节点。(0<n<=10^5)
第二行 n 个整数，依次表示每个节点的评分。(每个节点的评分不超过10^6)
接下来 n-1 行，每行 2 个整数 u, v，表示存在一条 u 到 v 的边。
由于这是一棵树，所以是不存在环的。
输出
输出一行一个数，表示上帝给这棵树的分数。
样例输入 Copy
5
1 -2 -3 4 5
4 2
3 1
1 2
2 5
样例输出 Copy
8

---

本题为树状dp，我们要求的是树中的最大连通块

对于一个无环的连通图，我们可以取任意的一点作为根节点来构建一棵树，然后搜索这棵树，找到最大的连通块
我们用f[u]表示以u为父节点包含u结点的所有连通块的最大值。
用w[i]来表示点i的权值

以题目中给的样例在这里放一颗树

我们要求f[1] 的最大值，就要求，所有子树中最大权值之和然后再加上w[i]，
如果向下再推， 我们就是要求f[2]的最大值，即求父节点为2的所有子树中最大权值之和然后再加上w[2],
然后，我们一直推下去。

所以我们就得到了本题的关键公式：
f[u] = w[u] + max(ff[1] + ff[2] + ···+f[u])（所有子树的最大权值和）

下面是代码：

#include <iostream>
#include <vector>

using namespace std;

typedef long long ll;

const int N = 100010;
int n;
int w[N];
ll f[N];
ll res;
vector<int> g[N];//用g来存下所有的边

void dfs(int u, int fa){
	for (int i = 0; i < g[u].size(); i ++ ){
		int son = g[u][i];
		if (son != fa){//遍历结点son的除父节点以外的所有结点，也就是遍历所有子树
			dfs(son, u);//接着向下一层子结点遍历
			f[u] += max(0ll, f[son]);
		}
	}
	
	res = max(res, f[u]);
}
int main(){
	cin >> n;
	for (int i = 1; i <= n; i ++ ) cin >> w[i], f[i] = w[i];
	for (int i = 0; i < n - 1; i ++ ){
		int u, v;
		scanf("%d%d", &u, &v);
		g[u].push_back(v);
		g[v].push_back(u); 
	}
	
	dfs(1, -1);
	
	cout << res << endl; 
	return 0;
}