BZOJ 5072 [Lydsy 十月月赛] 树DP 解题报告

Problem Statement

小A 成为了一个园艺家！他有一棵n 个节点的树（如果你不知道树是什么，请看Hint 部分）。他不小心打翻了墨水瓶，使得树的一些节点被染黑了。小A 发现这棵染黑了的树很漂亮，于是想从树中取出一个x 个点的联通子图，使得这些点中恰有y 个黑点，他想知道他的愿望能否实现。可是他太小，不会算，请
你帮帮他。

【解题报告】

考虑转化才成树上背包的形式，发现对于每一个x都有一个y可行的范围，直接判断是否在范围内就可以了。范围可以用背包DP求出
注意枚举的顺序

代码如下：

/**************************************************************
    Problem: 5072
    User: onepointo
    Language: C++
    Result: Accepted
    Time:10136 ms
    Memory:197092 kb
****************************************************************/

#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
#define inf 0x3f3f3f3f
#define N 5010

int t,n,q,cnt=-1;
struct Edge{int to,nxt;}e[N<<1];
int head[N],v[N],size[N],f[N][N],g[N][N];
int mn[N],mx[N];

void adde(int u,int v)
{
    e[++cnt].to=v;e[cnt].nxt=head[u];head[u]=cnt;
    e[++cnt].to=u;e[cnt].nxt=head[v];head[v]=cnt;
}
void dfs(int u,int fa)
{
    size[u]=1;
    f[u][1]=g[u][1]=v[u];
    for(int i=head[u];i!=-1;i=e[i].nxt)
    {
        int v=e[i].to;
        if(v==fa) continue;
        dfs(v,u);
        for(int j=size[u];j;j--)
        for(int k=0;k<=size[v];++k)
        {
            f[u][j+k]=max(f[u][j+k],f[u][j]+f[v][k]);
            g[u][j+k]=min(g[u][j+k],g[u][j]+g[v][k]);
        }
        size[u]+=size[v];
    }
    for(int i=1;i<=n;i++)
    {
        mx[i]=max(mx[i],f[u][i]);   
        mn[i]=min(mn[i],g[u][i]);
    }
}
int main()
{
    for(scanf("%d",&t);t;--t)
    {
        cnt=-1;
        memset(head,-1,sizeof(head));
        memset(g,inf,sizeof(g));
        memset(mn,inf,sizeof(mn));
        memset(f,0,sizeof(f));
        memset(mx,0,sizeof(mx));
        scanf("%d%d",&n,&q);
        for(int i=1;i<n;++i)
        {
            int u,v;scanf("%d%d",&u,&v);
            adde(u,v);
        }
        for(int i=1;i<=n;++i) scanf("%d",&v[i]);
        dfs(1,1);
        for(int i=1;i<=q;++i)
        {
            int x,y;scanf("%d%d",&x,&y); 
            puts(mn[x]<=y&&y<=mx[x]?"YES":"NO");
        }
        puts("");
    }
    return 0;
}