hdu 1520 树状dp 模板题。。

链接：http://acm.hdu.edu.cn/showproblem.php?pid=1520

题意：一堆人要聚会，为了避免不愉快，所以员工和他的直接上司不能同时出席，每个人出席的话都会有一个值，问怎么安排，是这个值最大~

思路：整个关系构成了一棵树，dp方程也很简单，看出来是树状dp的话就很简单了= =。。

#include<iostream>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<string>
#include<queue>
#include<algorithm>
#include<map>
#include<iomanip>
#define INF 99999999
using namespace std;//一堆头文件。。

const int maxn=6000+10;
int n,top,head[maxn],pre[maxn];//pre保存父节点是谁
int num[maxn];
int dp[maxn][2];

struct Side{//链式前向星存树（其实就是存图，只不过这个图是树）
    int v,next;
}side[maxn];

void init(){
    memset(head,-1,sizeof(head));
    memset(pre,0,sizeof(pre));
    memset(dp,0,sizeof(dp));
    top=0;
}

void add_side(int u,int v){
    side[top]=(Side){v,head[u]};
    head[u]=top++;
}

void dfs(int u){//搜索 答案
    for(int i=head[u];i != -1;i=side[i].next){
        int v=side[i].v;
        dfs(v);
        dp[u][0] += max(dp[v][0],dp[v][1]);
        dp[u][1] += dp[v][0];
    }
    dp[u][1] += num[u];
}

int main(){
    int u,v;
    while(~scanf("%d",&n)){
        init();
        for(int i=1;i<=n;++i)
            scanf("%d",&num[i]);
        while(~scanf("%d%d",&v,&u),u+v)
            pre[v]=u;
        for(int i=1;i<=n;++i)
            add_side(pre[i],i);
        dfs(0);//假设0节点是大boss的“上司”
        printf("%d\n",dp[0][0]);
    }
    return 0;
}