树形dp 入门

以前看过点树形dp，不过全忘了。今天做了一道入门级的简单题，回忆了一下。所谓树形dp就是在一棵树上进行状态转移。有时候他的状态转移方程比普通dp还简单，只不过建图比较麻烦，顺便复习了一下邻接表的用法，真是好久不练就都忘了。

下面是一道最基础的题，要求父节点和儿子节点不能同时选，求最大权值，很明显状态转移方程就是dp[i][1]+=dp[j][0]; dp[i][0]+=max(dp[j][0],dp[j][1])

其中0表示不选，1表示选。进行dp其实就是一个递归的过程，先求出儿子节点的结果，再去更新父节点。

#include<stdio.h>
#include<string.h>
int dp[100010][2];
int head[100010];
struct s{
    int to,next;
};
s edg[200010];
int cnt;
void add(int u,int v){
    edg[cnt].to=v;
    edg[cnt].next=head[u];
    head[u]=cnt++;
    edg[cnt].to=u;
    edg[cnt].next=head[v];
    head[v]=cnt++;
}
void dfs(int x,int y){
    for(int i=head[x];i!=-1;i=edg[i].next){
        int v=edg[i].to;
        if(v==y) continue;
        dfs(v,x);
        dp[x][1]+=dp[v][0];
        dp[x][0]+=dp[v][0]>dp[v][1]?dp[v][0]:dp[v][1];
    }
}
int main(){
    int n;
    cnt=0;
    scanf("%d",&n);
    int u,v;
    memset(dp,0,sizeof(dp));
    memset(head,-1,sizeof(head));
    for(int i=1;i<=n;i++)scanf("%d",&dp[i][1]);
    for(int i=1;i<n;i++){
        scanf("%d%d",&u,&v);
        add(u,v);
    }
    dfs(1,-1);
    printf("%d\n",dp[1][0]>dp[1][1]? dp[1][0]:dp[1][1]);
    return 0;
}