HDU 4616 Game （搜索）、（树形dp）

题目地址：http://acm.hdu.edu.cn/showproblem.php?pid=4616

这道题目数据可能比较弱，搜索都可以AC，但是不敢写，哎……

搜索AC代码：

#include <iostream>
#include <cstdio>
#include <cstring>
#include <string>
#include <cstdlib>
#include <cmath>
#include <vector>
#include <list>
#include <deque>
#include <queue>
#include <iterator>
#include <stack>
#include <map>
#include <set>
#include <algorithm>
#include <cctype>
#include <ctime>
#pragma comment(linker, "/STACK:16777216")
using namespace std;

typedef __int64 LL;
const int N=50005;
const int M=55555555;
const int INF=0x3f3f3f3f;
const double PI=acos(-1.0);

vector<int>q[N];
int xh[N],w[N];
int n,c,sm;
bool vis[N];

struct love
{
    int i,sum;
    int cc;
}r,e,my[M];

void BFS(int s)
{
    memset(vis,0,sizeof(vis));
    vis[s]=1;
    vector<int>::iterator it;
    r.i=s;
    r.sum=xh[s];    sm=max(sm,r.sum);
    r.cc=c-w[s];
    if(r.cc==0)
        return ;
    int he=0,ta=0;
    my[ta++]=r;
    while(he!=ta)
    {
        e=my[he++];
        if(he==M)
            he=0;
        for(it=q[e.i].begin();it!=q[e.i].end();it++)
        {
            r=e;
            int k=*it;
            if(vis[k])  continue;
            vis[k]=1;
            r.i=k;
            r.sum+=xh[k];   sm=max(sm,r.sum);
            r.cc-=w[k];
            if(r.cc>0)
            {
                my[ta++]=r;
                if(ta==M)
                    ta=0;
            }
        }
    }
}

int main()
{
    int T;
    scanf("%d",&T);
    while(T--)
    {
        scanf("%d%d",&n,&c);
        for(int i=0;i<n;i++)
        {
            scanf("%d%d",&xh[i],&w[i]);
        }
        for(int i=0;i<=n;i++)
            q[i].clear();
        for(int i=0;i<n-1;i++)
        {
            int a,b;
            scanf("%d%d",&a,&b);
            q[a].push_back(b);
            q[b].push_back(a);
        }
        int Max=-1;
        for(int i=0;i<n;i++)
        {
            sm=0;
            if(q[i].size()==1)
                BFS(i);
            if(Max<sm)
                Max=sm;
        }
        cout<<Max<<endl;
    }
    return 0;
}

树形DP代码：

#include <iostream>
#include <cstdio>
#include <vector>
#include <cstring>
using namespace std;
#define N 50500
int trap[N],val[N],n,C,vis[N],ans;
int dp[N][5][2];
vector<int>vec[N];

void DP(int u)
{
    vis[u]=1;
    dp[u][trap[u]][0]=dp[u][trap[u]][1]=val[u];
    int v;
    for(int i=0;i<vec[u].size();++i)
    {
        v=vec[u][i];
        if(vis[v]) continue;
        DP(v);
        for(int j=0;j<=C;++j)
            for(int k=0;j+k<=C;++k)
        {
            if(j!=C) // j=C的时候已经停止了，不会再往子节点跑了
                ans=max(ans,dp[u][j][0]+dp[v][k][1]);
            if(k!=C) // 同理
                ans=max(ans,dp[u][j][1]+dp[v][k][0]);
            if(j+k<C) // 0表示从子树方向往根节点跑，子树可能是非陷阱的点，
                //假如j+k==C，题意要求到达C要停止，但按dp[v][k][0]记录的，还会继续跑，所以不行
                ans=max(ans,dp[u][j][0]+dp[v][k][0]);
            //if(j+k<C)
                //ans=max(ans,dp[u][j][1]+dp[v][k][1]);
        }
        for(int j=0;j+trap[u]<=C;++j)
            dp[u][j+trap[u]][0]=max(dp[u][j+trap[u]][0],dp[v][j][0]+val[u]);
        for(int j=1;j+trap[u]<=C;++j) // 1表示从根节点往下走的，1->0是不可取的，因为算上1后，已经满C了，就不会再走了，所以不能更新
            dp[u][j+trap[u]][1]=max(dp[u][j+trap[u]][1],dp[v][j][1]+val[u]);
    }
}
int main ()
{
    int ncase;scanf("%d",&ncase);
    while(ncase--)
    {
        scanf("%d%d",&n,&C);
        for(int i=0;i<n;++i)
            vec[i].clear();
        memset(vis,0,sizeof(vis));
        memset(dp,0,sizeof(dp));
        ans=0;
        for(int i=0;i<n;++i)
            scanf("%d%d",&val[i],&trap[i]);
        int u,v;
        for(int i=1;i<n;++i)
        {
            scanf("%d%d",&u,&v);
            vec[u].push_back(v);
            vec[v].push_back(u);
        }
        DP(0);
        printf("%d\n",ans);
    }
    return 0;
}