Anniversary party------树形dp

第一次写这种树形dp，尴尬，然后这里用链式向前星会比一般的快很多，然后就没什么要说的了，看看代码就懂了。

#include<cstdio>
#include<cstring>
#include<iostream>
using namespace std;
const int MX=1e5+9;
int n,dp[2][MX],st,u,v,pos[MX];

struct node{
    int to,next;
}edge[MX];
int head[MX],tot=0;
void add(int u,int v){
    edge[tot].to=v;
    edge[tot].next=head[u];
    head[u]=tot++;
}

void dfs(int st){
    for( int i=head[st] ; ~i ; i=edge[i].next ){
        int v=edge[i].to;
        dfs(v);
        dp[0][st]=max(dp[0][st]+dp[0][v],dp[0][st]+dp[1][v]);
        dp[1][st]=max(dp[1][st],dp[1][st]+dp[0][v]);
    }
}

int main()
{
  //  freopen("input.txt","r",stdin);
    memset(head,-1,sizeof(head));
    memset(pos,0,sizeof(pos));
    memset(dp,0,sizeof(dp));
    scanf("%d",&n);
    for( int i=1 ; i<=n ; i++ )
        scanf("%d",&dp[1][i]);
    for( int i=1 ; i<=n-1 ; i++ ){
        scanf("%d %d",&v,&u);
        pos[v]=1;
        add(u,v);
    }
    for( int i=1 ; i<=n ; i++ )
        if( pos[i]==0 )
            st=i;
    dfs(st);
    printf("%d\n",max(dp[0][st],dp[1][st]));
    return 0;
}