poj2342Anniversary party（树形dp）

树形dp入门题；

树形dp一般就是确定dp的维数，然后一个dfs就完了，相当于树上递推，从底层递推到上一层

dp[u][0]+=max(dp[v][1],dp[v][0]); 上司u不来，下司v可以来可以不来

dp[u][1]+=dp[v][0];上司u来

#include<algorithm>
#include<stdio.h>
#include<string.h>
#include<vector>
#include<iostream>
using namespace std;
const int N=6010;
int dp[N][2],head[N<<1],cnt,n,val[N];
int in[N],ou[N];
struct node
{
    int u,v,w,next;
}ed[N<<2];
void add(int u,int v,int w)
{
    ed[cnt].u=u;
    ed[cnt].v=v;
    ed[cnt].w=w;
    ed[cnt].next=head[u];
    head[u]=cnt++;
}
void dfs(int u)
{
    for(int i=head[u];i!=-1;i=ed[i].next)
    {
        int v=ed[i].v;
        int w=ed[i].w;
        dfs(v);
        dp[u][0]+=max(dp[v][1],dp[v][0]);
        dp[u][1]+=dp[v][0];
    }
}
int main()
{
    memset(head,-1,sizeof(head));
    cnt=0;
    scanf("%d",&n);
    for(int i=1;i<=n;i++)
    {
        scanf("%d",&val[i]);
    }
    int u,v;
    for(int i=1;i<=n-1;i++)
    {
        scanf("%d%d",&u,&v);
        add(v,u,val[v]);
        in[u]++;
        ou[v]++;
    }
    cin>>u>>v;
    memset(dp,0,sizeof(dp));
    int mr;
    for(int i=1;i<=n;i++)
    {
        dp[i][1]=val[i];
        if(in[i]==0)
            mr=i;
    }
    dfs(mr);
    printf("%d\n",max(dp[mr][0],dp[mr][1]));
    return 0;
}
//