POJ2342_Anniversary party 树形DP

 题意： 给定一个n，　对于每个人有一个权重，然后给定一个上下级关系，上级来了直属下级不能取，最后问最大权重值。

做法：　树形ＤＰ，先确定根节点，然后在通过转移方程

   dp[root][1]+=dp[son][0];
    dp[root][0]+=max(dp[son][0],dp[son][1]); 确定

AC 代码：

#include<iostream>
#include<vector>
#include<cstring>
using namespace std;
int p[6001];
int dp[6001][2];
vector<int>edge[6011];
int num[6011];
int father[6011];
int work(int root)
{
    dp[root][0]=0;dp[root][1]=p[root];
    for(int i=0;i<num[root];i++)
    {
        int temp=edge[root][i];
        //cout<<temp<<endl;
        work(temp);
        dp[root][1]+=dp[temp][0];
        dp[root][0]+=max(dp[temp][0],dp[temp][1]);
    }
    return max(dp[root][0],dp[root][1]);
}
int main()
{
    int n;
    cin>>n;
    memset(p,0,sizeof(p));
    memset(dp,0,sizeof(dp));
    memset(num,0,sizeof(num));
    memset(father,0,sizeof(father));
    for(int i=1;i<=n;i++)
    cin>>p[i];

    int a,b;
    while(cin>>a>>b)
    {
        if(a==0&&b==0)break;
        else
        {
            edge[b].push_back(a);
            father[a]=1;
            num[b]++;
        }
    }
    int root;
    for(int i=1;i<=n;i++){if(father[i]==0) root=i;}
 //   cout<<"root is "<<root<<endl;
    int Max=work(root);
    cout<<Max<<endl;


}