题意(翻译来自hust vj)
一棵树有n个结点,每个结点有一个权值,一个结点和其直接的父亲结点不能同时选,求按照要求所能选到的点权和的最大值
解法
树形DP
设dp[i][0]表示不取i节点的i子树的最大权值和,dp[i][1]表示取i节点的i子树的最大权值和
设v是u的子节点,则
dp[u][0] = Sum(max(dp[v][0], dp[v][1]))
dp[u][1] = v[u] + Sum(dp[v][0])
边界条件,即当节点u为叶节点时
dp[u][0] = 0
dp[u][1] = v[u]
代码
#include <cstdio>
#include <cstring>
#include <algorithm>
#include <vector>
using namespace std;
const int N = 6010;
int n;
int fat[N]; //父节点
int v[N]; //权值
int dp[N][2]; //dp[i][0]表示i节点不选,dp[i][1]表示i节点选
vector<int> son[N]; //子节点
void BuildTree(int u) {
if (son[u].size() == 0) { //若是叶节点
dp[u][0] = 0; //不取时,dp[u][0] = 0
dp[u][1] = v[u]; //取时,dp[u][1] = v[u]
} else {
dp[u][0] = 0; //不取时,dp[u][0] = 0
dp[u][1] = v[u]; //取时,dp[u][1] = v[u]
for (int i = 0; i < (int)son[u].size(); i++) {
int v = son[u][i];
BuildTree(v);
dp[u][0] += max(dp[v][0], dp[v][1]);
dp[u][1] += dp[v][0];
}
}
}
int main() {
while (scanf("%d", &n) != EOF) {
//判断是否为结束标志 0 0
if (n == 0) {
scanf("%d", &n);
if (n == 0) break;
}
for (int i = 1; i <= n; i++) {
scanf("%d", &v[i]); //读入节点权值
son[i].clear(); //清空子节点列表
}
memset(fat, 0, sizeof(fat)); //清空父节点
for (int i = 1; i < n; i++) {
int a, b; scanf("%d%d", &a, &b);
fat[a] = b; //读入父节点
son[b].push_back(a); //添加子节点
}
//寻找根节点
int root = 0;
for (int i = 1; i <= n; i++) {
if (fat[i] == 0) {
root = i;
break;
}
}
BuildTree(root);
printf("%d\n", max(dp[root][0], dp[root][1]));
}
return 0;
}