bzoj 5072 小A的树 —— 树形DP

题目：https://www.lydsy.com/JudgeOnline/problem.php?id=5072

由于对于一个子树，固定有 j 个黑点，连通块大小是一个连续的范围；

所以记 f[i][j] 表示以 i 为根的子树中选 j 个黑点，连通块最大的点数，g[i][j] 表示最小的点数；

然后普通树形DP即可，注意初始化；

但怎么处理询问？这道题卡空间，只能开 2.5 个 5000*5000 的 int 数组；

其实，对于整棵树，固定有 j 个黑点的连通块大小也是一个连续的范围，所以每个子树的答案直接贡献到 0 的答案上，就可以 O(1) 判断了。

代码如下：

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
int const xn=,inf=0x3f3f3f3f,base=;
int T,n,q,f[xn][xn],g[xn][xn],hd[xn],ct,nxt[xn<<],to[xn<<],siz[xn],tf[xn],tg[xn];
bool b[xn];
void add(int x,int y){to[++ct]=y; nxt[ct]=hd[x]; hd[x]=ct;}
int rd()
{
  int ret=,f=; char ch=getchar();
  while(ch<''||ch>''){if(ch=='-')f=; ch=getchar();}
  while(ch>=''&&ch<='')ret=(ret<<)+(ret<<)+ch-'',ch=getchar();
  return f?ret:-ret;
}
void dfs(int x,int fa)
{
  siz[x]=b[x];
  if(b[x])f[x][]=g[x][]=;//
  else f[x][]=,g[x][]=;//
  for(int i=hd[x],u;i;i=nxt[i])
    {
      if((u=to[i])==fa)continue;
      dfs(u,x);
      memcpy(tf,f[x],sizeof f[x]);
      memcpy(tg,g[x],sizeof g[x]);
      for(int j=siz[x]+siz[u];j>=b[x];j--)
    for(int k=;k<=min(j,siz[u]);k++)
      if(j-k<=siz[x])
        {
          f[x][j]=max(f[x][j],tf[j-k]+f[u][k]);
          g[x][j]=min(g[x][j],tg[j-k]+g[u][k]);
        }
      siz[x]+=siz[u];
    }
  if(!b[x])g[x][]=;
  else f[x][]=g[x][]=;
  for(int j=;j<=siz[x];j++)f[][j]=max(f[][j],f[x][j]),g[][j]=min(g[][j],g[x][j]);
}
int main()
{
  T=rd();
  while(T--)
    {
      n=rd(); q=rd(); ct=;
      memset(hd,,sizeof hd);
      memset(f,-,sizeof f); memset(g,0x3f,sizeof g);
      for(int i=,x,y;i<n;i++)
    {
      x=rd(); y=rd();
      add(x,y); add(y,x);
    }
      for(int i=;i<=n;i++)b[i]=rd();
      dfs(,);
      for(int i=,x,y;i<=q;i++)
    {
      x=rd(); y=rd();
      if(x>=g[][y]&&x<=f[][y])puts("YES");
      else puts("NO");
    }
      puts("");
    }
  return ;
}