题目:https://www.lydsy.com/JudgeOnline/problem.php?id=5072
发现对于每个子树,黑点个数确定时,连通块的大小取值范围一定是一段区间;所以考虑只最小化最小值、最大化最大值,记 f 和 g 简单dp即可。
注意可能从当前子树里选0个点!此时会用自己更新自己!!所以要先复制一份原来的用来更新!
快速回答询问,本可以记差分数组,每个子树算完后给合法部分区间赋值;但空间开不下。
于是绞尽脑汁,终于想出可以开 bool 数组分块来赋值!!然而WA得不行。
交流后发现那个“取值是一段区间”的性质,在全局也是适用的(很明显……)!所以只要更新一下合法的最值就行了……
然后对拍半天 bool 分块,发现因为有 0 的值,所以标号应该是 0~base-1 这样;而且代码里注释的那个部分是 < 和 >= 而不是 <= 和 >!
然后又因为数据生成出错而以为自己还是不对而又拍、查了半天;最后分块的方法虽然慢一点,但也A了。感觉很好!
#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<cmath>
using namespace std;
const int N=,M=;
int T,n,q,base,hd[N],xnt,to[N<<],nxt[N<<],siz[N],f[N][N],g[N][N],tf[N],tg[N];
bool b[N],ok[N][N],ok2[M][N];//siz,black
int rdn()
{
int ret=;bool fx=;char ch=getchar();
while(ch>''||ch<''){if(ch=='-')fx=;ch=getchar();}
while(ch>=''&&ch<='') ret=(ret<<)+(ret<<)+ch-'',ch=getchar();
return fx?ret:-ret;
}
void add(int x,int y)
{
to[++xnt]=y;nxt[xnt]=hd[x];hd[x]=xnt;
to[++xnt]=x;nxt[xnt]=hd[y];hd[y]=xnt;
}
int calc(int x){return x/base+;}//0~base-1
void dfs(int cr,int fa)
{
f[cr][b[cr]]=g[cr][b[cr]]=; siz[cr]=b[cr];
for(int i=hd[cr],v;i;i=nxt[i])
if((v=to[i])!=fa)
{
dfs(v,cr);
memcpy(tf,f[cr],sizeof f[cr]);
memcpy(tg,g[cr],sizeof g[cr]);
for(int j=siz[cr]+siz[v];j>=b[cr];j--)
for(int k=max((int)b[v],j-siz[cr]);k<=siz[v]&&k<=j;k++)
{
f[cr][j]=min(f[cr][j],tf[j-k]+f[v][k]);
g[cr][j]=max(g[cr][j],tg[j-k]+g[v][k]);
}
siz[cr]+=siz[v];
}
/*
for(int i=0;i<=siz[cr];i++)
f[0][i]=min(f[0][i],f[cr][i]),
g[0][i]=max(g[0][i],g[cr][i]);
*/
for(int i=;i<=siz[cr];i++)
{
if(f[cr][i]>g[cr][i])continue;
int l=calc(f[cr][i]),r=calc(g[cr][i]);
if(r-l<=)
{
for(int j=f[cr][i];j<=g[cr][i];j++)
ok[j][i]=;
}
else
{
for(int j=l+;j<r;j++)ok2[j][i]=;
int lm=l*base;
for(int j=f[cr][i];j<lm;j++)ok[j][i]=;//<
lm=(r-)*base;
for(int j=g[cr][i];j>=lm;j--)ok[j][i]=;//>=
}
}
}
int main()
{
T=rdn();
while(T--)
{
n=rdn(); q=rdn(); base=sqrt(n);
memset(hd,,sizeof hd); xnt=;
memset(ok,,sizeof ok); memset(ok2,,sizeof ok2);
for(int i=,u,v;i<n;i++)
{
u=rdn(); v=rdn(); add(u,v);
}
for(int i=;i<=n;i++)b[i]=rdn();
memset(f,0x3f,sizeof f); memset(g,-,sizeof g);
dfs(,);
for(int i=,x,y,d;i<=q;i++)
{
x=rdn(); y=rdn(); d=calc(x);
puts((ok[x][y]||ok2[d][y])?"YES":"NO");
//puts(x>=f[0][y]&&x<=g[0][y]?"YES":"NO");
}
puts("");
}
return ;
}