第一把树形dp , 不懂树形dp,百度学习,这是树形dp入门,对着别人的代码打的,然后懂了。
题意:有一个公司的人要去参加party,每个人有一个愉悦值。每个人和他的直接上司不能同时出现。要求邀请一定量的人使得愉悦值最大。
树形dp,对于第i个人,有两种选择,选或不选。dp[i][0] , dp[i][1];
推出dp方程dp[i][0] += max(dp[st][1],dp[st][0]) st为i的所用子树。
dp[i][1] += d[st][0]; st为子树。
对根节点做一遍的dfs就可以求出最大值了。 代码更加容易懂,上代码。
#include <iostream>
#include <cstdio>
#include <cmath>
#include <queue>
#include <algorithm>
#include <vector>
#include <cstring>
using namespace std;
#define maxn 10010
int dp[maxn][2];
int father[maxn],visit[maxn];
int n;
void dfs(int root) {
visit[root] = 1;
for(int i = 1; i <= n; i++) {
if(visit[i] == 0 && father[i] == root) {
dfs(i);
dp[root][0] += max(dp[i][0],dp[i][1]);
dp[root][1] += dp[i][0];
}
}
}
int main() {
while(~scanf("%d",&n)) {
memset(dp,0,sizeof(dp));
memset(visit,0,sizeof(visit));
memset(father,0,sizeof(father));
for(int i = 1; i <= n; i++)
scanf("%d",&dp[i][1]);
int x,f,root;
while(scanf("%d %d",&x,&f),x||f) {
father[x] = f;
}
root = x;
while(father[root]) root = father[root];
dfs(root);
printf("%d\n",max(dp[root][0],dp[root][1]));
}
return 0;
}