HDU 1520 Anniversary party （简单树形DP入门）

树形dp，可能是涉及到图论和dfs，显得比较高大上，我现在只做了一题，也没有什么发言权。目前总结如下：

1.“树”，当作有向图（而且还无环的）来处理，建图（树）的时候更为简单，只要在它和它的孩子节点之间加一条指向孩子的有向边就可以了。

2.dp还是一样的dp，但是树形dp往往不是“线性、水平”的了，而是要从叶子节点推到根节点（或者反过来），关系盘结的更为复杂。

这题是一道最简单的树形dp；方程也很容易想到

当i来的时候，dp[i][1] += dp[j][0];//j为i的下属

当i不来的时候，dp[i][0] +=max(dp[j][1],dp[j][0]);//j为i的下属

这里我是用vector实现的邻接表，因为是有向图，每个节点只会被访问一次，甚至连vis数组都可以不用了。用结构体也是一样。

【代码】

/* ***********************************************
Author        :angon
2016年8月15日 23:52:07
************************************************ */
#include <stdio.h>
#include <string.h>
#include <iostream>
#include <algorithm>
#include <stack>
#include <vector>
#include <queue>
#include <set>
#include <map>
#include <string>
#include <math.h>
#include <stdlib.h>
#include <time.h>
using namespace std;
#define showtime fprintf(stderr,"time = %.15f\n",clock() / (double)CLOCKS_PER_SEC)
#define lld %I64d
#define REP(i,k,n) for(int i=k;i<n;i++)
#define REPP(i,k,n) for(int i=k;i<=n;i++)
#define scan(d) scanf("%d",&d)
#define scanl(d) scanf("%I64d",&d)
#define scann(n,m) scanf("%d%d",&n,&m)
#define scannl(n,m) scanf("%I64d%I64d",&n,&m)
#define mst(a,k)  memset(a,k,sizeof(a))
#define LL long long
#define N 6005
#define mod 1000000007
inline int read(){int s=0;char ch=getchar();for(; ch<'0'||ch>'9'; ch=getchar());for(; ch>='0'&&ch<='9'; ch=getchar())s=s*10+ch-'0';return s;}

int n,dp[N][2],father[N];
vector<int>G[N];
void dfs(int root)
{
    for(int i=0;i<G[root].size();i++)
    {
        dfs(G[root][i]);  //递归root的每个孩子
    }
    for(int i=0;i<G[root].size();i++) //回溯时计算
    {
        dp[root][0] += max(dp[G[root][i]][0],dp[G[root][i]][1]);
        dp[root][1] += dp[G[root][i]][0];
    }
}
int main()
{
    //freopen("in.txt","r",stdin);
    //freopen("out.txt","w",stdout);
    int u,v;
    while(~scan(n))
    {
        mst(father,-1); mst(dp,0);
        REPP(i,1,n) { scan(dp[i][1]); G[i].clear();}
        while(scann(u,v)&&u&&v)
        {
            father[u]=v;
            G[v].push_back(u);
        }
        int root=1;
        while(father[root]!=-1) root=father[root];
        dfs(root);
        printf("%d\n",max(dp[root][1],dp[root][0]));
    }

    return 0;
}