SGU 143.Long Live the Queen（女王万岁）

时间限制：0.25s

空间限制：4M

题意：

    有n(n<=16000)个小镇，每两个小镇有且仅有一条路径相连。每个小镇有一个收益x（-1000<=x<=1000）.
    现在要求,选择一些小镇,满足下面两点要求:
    1.选择的小镇都能互相到达,并且路径上的小镇也被选择了.
    2.选择的小镇的收益和最大.

输入

   一个整数n,接下来n个整数,代表这n个小镇的收益.
   接下来n-1行,每行两个整数代表这两个小镇有道路直接相连.

输出

   一个整数,代表最大的收益

Sample Input

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

Sample Output

4

 

 

 

 

---

 

Solution:

                先来分析一下题目模型：

 

             有一颗n(n<=16000)个节点的树，每个节点都有一个权值k（-1000<=k<=1000）,

 

             求这颗树的一颗子树,使得子树的所有节点的权值和最大.

 

             假设已经选择了一些点,对于于他相邻的点,求出相邻点的连通块的和的最大值,如果这个值大于0,那么久把这个连通块加入选择.

             DFS即可.

 

参考代码:

 

 

#include <iostream>
#include <cstring>
#include <cstdio>
#include <vector>
#define INf 16666
#define Maxn 0xfffffff
using namespace std;
vector<int> g[INf];
int val[INf], f[INf];
int n, x, y, tem;
int dfs (int x, int fa) {
	f[x] = val[x];
	for (int i = 0; i < g[x].size(); i++) {
              tem=0;
		if (g[x][i] != fa)
			tem = dfs (g[x][i], x);
		if (tem > 0) f[x] += tem;
	}
	return f[x];
}
int main() {
	scanf ("%d", &n);
	for (int i = 1; i <= n; i++) scanf ("%d", &val[i]);
	for (int i = 1; i < n; i++) {
		scanf ("%d %d", &x, &y);
		g[x].push_back (y);
		g[y].push_back (x);
	}
	dfs (1, -1);
	int ans = -Maxn;
	for (int i = 1; i <=n; i++)
		ans = max (ans, f[i]);
	printf ("%d", ans);
}