hdu4616 Game 【树形dp】

题解详见：http://blog.csdn.net/martinue/article/details/51025232
分析透彻。

#include<iostream>
#include<cstdio>
#include<cstring>
#include<string>
#include<algorithm>
#include<cmath>
#include<vector>
#include<queue>
#include<set>
#define ll long long
using namespace std;

int getint()
{
    int i=0,f=1;char c;
    for(c=getchar();(c<'0'||c>'9')&&c!='-';c=getchar());
    if(c=='-')f=-1,c=getchar();
    for(;c>='0'&&c<='9';c=getchar())i=(i<<3)+(i<<1)+c-'0';
    return i*f;
}

const int N=50005,INF=0x3f3f3f3f;
int T,n,C,ans;
int val[N],flag[N],dp[N][4][2];
int tot,first[N],nxt[N<<1],to[N<<1];

void add(int x,int y)
{
    nxt[++tot]=first[x],first[x]=tot,to[tot]=y;
}

void dfs(int u,int fa)
{
    dp[u][flag[u]][flag[u]]=val[u];
    for(int e=first[u];e;e=nxt[e])
    {
        int v=to[e];
        if(v==fa)continue;
        dfs(v,u);
        for(int i=0;i<=C;i++)
            for(int j=0;i+j<=C;j++)
            {
                if(i+j<=C)ans=max(ans,dp[u][i][1]+dp[v][j][1]);
                if(i+j<C) ans=max(ans,dp[u][i][0]+dp[v][j][0]);
                if(i!=0)  ans=max(ans,dp[u][i][1]+dp[v][j][0]);
                if(j!=0)  ans=max(ans,dp[u][i][0]+dp[v][j][1]);
            }
        for(int i=0;i<C;i++)
        {
            dp[u][i+flag[u]][0]=max(dp[u][i+flag[u]][0],dp[v][i][0]+val[u]);
            dp[u][i+flag[u]][1]=max(dp[u][i+flag[u]][1],dp[v][i][1]+val[u]);
        }
        if(!flag[u])dp[u][C][1]=max(dp[u][C][1],dp[v][C][1]+val[u]);
    }
}

int main()
{
    //freopen("lx.in","r",stdin);
    int x,y;
    T=getint();
    while(T--)
    {
        memset(dp,-INF,sizeof(dp));
        memset(first,0,sizeof(first));
        ans=-INF,tot=0;
        n=getint(),C=getint();
        for(int i=1;i<=n;i++)
            val[i]=getint(),flag[i]=getint();
        for(int i=1;i<n;i++)
        {
            x=getint()+1,y=getint()+1;
            add(x,y),add(y,x);
        }
        dfs(1,0);
        cout<<ans<<'\n';
    }
    return 0;
}