蓝桥杯 ALGO-4 结点选择

  算法训练 结点选择  

时间限制：1.0s   内存限制：256.0MB

      

问题描述

有一棵 n 个节点的树，树上每个节点都有一个正整数权值。如果一个点被选择了，那么在树上和它相邻的点都不能被选择。求选出的点的权值和最大是多少？

输入格式

第一行包含一个整数 n 。

接下来的一行包含 n 个正整数，第 i 个正整数代表点 i 的权值。

接下来一共 n-1 行，每行描述树上的一条边。

输出格式

输出一个整数，代表选出的点的权值和的最大值。

样例输入

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

样例输出

12

样例说明

选择3、4、5号点，权值和为 3+4+5 = 12 。

数据规模与约定

对于20%的数据， n <= 20。

对于50%的数据， n <= 1000。

对于100%的数据， n <= 100000。

权值均为不超过1000的正整数。

分析：最开始想使用贪心，每次遍历找出最大值并将其余其相邻节点去除，由此找出最大和。失败后转用树状DP，思想为：若选择某节点，则不选择其相邻节点；若不选择某节点，则可选择也可不选其相邻节点。dp数组的含义横坐标为节点编号，纵坐标为是否选择此节点。

代码：

#include<iostream>
#include<vector>
#include<queue>
using namespace std;
#define MAXN 100001
struct Node {
	int num;
	int weight;
	vector<int>	adj;
}G[MAXN];
int visit[MAXN] = { 0 };
int dp[MAXN][2] = { 0 };
void dfs(int start) {
	visit[start] = 1;
	dp[start][0] = 0;
	dp[start][1] = G[start].weight;
	for (int i = 0; i < G[start].adj.size(); i++) {
		int v = G[start].adj[i];
		if (visit[v] == 0) {
			dfs(v);
			dp[start][0] += max(dp[v][0], dp[v][1]);
			dp[start][1] += dp[v][0];
		}
	}
}
int main() {
	int n;
	cin >> n;
	for (int i = 1; i <= n; i++) {
		G[i].num = i;
		cin >> G[i].weight;
	}
	for (int i = 0; i < n - 1; i++) {
		int u, v;
		cin >> u >> v;
		G[u].adj.push_back(v);
		G[v].adj.push_back(u);
	}
	dfs(1);
	cout << max(dp[1][0], dp[1][1]) << endl;;
	return 0;
}