树形DP- Anniversary party

树形DP专题已经过去了一星期了。写几道题解来加强理解吧。

Anniversary party

 （树形DP入门）

题意：

有n个人要去参加party，但是一个人来参加了，那么他的下属就不能来。输入的时候，n行给出n个人来的价值，然后输入x，y，表示x是y的下属，x和y同时为0输入停止，表示x是y的下属。求这次party可以获得的最大价值。

思路：树形DP，dp【i】【0】表示第i个人不来，dp【i】【1】表示第i个人来。那么状态转移方程就是

dp[f][1]+=dp[z][0];

dp[f][0]+=max(dp[z][0],dp[z][1]);

z为f的下属。

AC代码：

#include<iostream>
#include<cmath>
#include<cstring>
#include<algorithm>
#include<cstdio>
#include<string>
#include<queue>
#include<vector>
#include<map>
using namespace std;
const int mx=100010;
struct node{
    int w,v,nex;
}a[mx<<1];
int dp[mx][3],h[mx],n,t;
int f[mx];
void add(int w,int v,int f)
{
 a[t].w=w;
 a[t].v=v;
 a[t].nex=h[f];
 h[f]=t++;
}
void dfs1(int f);
int main()
{
    while(scanf("%d",&n)!=EOF)
    {
        memset(h,-1,sizeof(h));
        memset(f,-1,sizeof(f));
        memset(dp,0,sizeof(dp));t=0;
        for(int i=1;i<=n;i++)
        {scanf("%d",&dp[i][1]);}
         int x,y;
         while (scanf("%d%d",&x,&y)&&!(x==0&&y==0))
       { add(x,x,y);f[x]=y;}
        int i=1;
        while(f[i]!=-1) i=f[i];
        dfs1(i);
        printf("%d\n",max(dp[i][1],dp[i][0]));
    }
return 0;
}
void dfs1(int f)
{
    for(int i=h[f];i!=-1;i=a[i].nex)
    {
        int z=a[i].v;
        dfs1(z);
        {
         dp[f][1]+=dp[z][0];
         dp[f][0]+=max(dp[z][0],dp[z][1]);
        }
    }
}