HDU_1520_Anniversary party_树形DP

题目：给你一颗有根树，每个结点附带一个权值（正整数），问你：从中选出一些点，使得权值之和最大，但要求父节点和子节点只能选一个，输出该最大权值和。

解题思路：

基本树形DP，dp[i][j]表示以i为根的子树可获得的最大权值和，j=0表示不选i结点，j=1表示选i结点；

递推公式 dp[father][0]+=max(dp[son][0],dp[son][1])(父亲不去，儿子去或不去取最大的）

dp[father][1]+=dp[son][0]+weight[father];(父亲去了，儿子必须去）

代码如下：

#include<iostream>
#include<vector>
#include<algorithm>
using namespace std;


const int MAX=6001;
int father[MAX],dp[MAX][2],a[MAX];
vector<int> e[MAX];


inline void dfs(int cur)
{
dp[cur][1]=a[cur];
for(int i=0;i<e[cur].size();i++)
{
int son=e[cur][i];
dfs(son);
dp[cur][0]+=max(dp[son][0],dp[son][1]);
dp[cur][1]+=dp[son][0];
}
return;
}


int main()
{
//freopen("test.txt","r",stdin);
int n;
while(~scanf("%d",&n))
{
for(int i=1;i<=n;i++)
{
cin>>a[i];
father[i]=-1;
e[i].clear();
dp[i][0]=dp[i][1]=0;
}
int a,b;
while(~scanf("%d%d",&a,&b),a+b)
{
father[a]=b;
e[b].push_back(a);
}
int rt=1;
while(father[rt]!=-1) rt=father[rt];
dfs(rt);
printf("%d\n",max(dp[rt][1],dp[rt][0]));
}
return 0;
}