Description
BZOJ只是扔了个下载链接
Solution
设$f[x][i]$表示$x$点选中$i$个黑点的最小连通块。
设$g[x][i]$表示$x$点选中$i$个黑点的最大连通块。
转移非常明显。处理出每个情况的上下界之后差分一下$O(1)$回答询问即可。
卡空间所以要用$short$。 第二次$INF$开大了……应该多上点心了……
Code
#include<iostream>
#include<cstring>
#include<cstdio>
#define N (5009)
using namespace std; struct Edge{int to,next;}edge[N<<];
int n,u,v,T,q,INF=1e4,size[N],b[N];
short f[N][N],g[N][N],tmpf[N],tmpg[N],ans[N][N];
int head[N],num_edge; void add(int u,int v)
{
edge[++num_edge].to=v;
edge[num_edge].next=head[u];
head[u]=num_edge;
} void Dfs(int x,int fa)
{
size[x]=;
for (int i=; i<=n; ++i)
f[x][i]=INF,g[x][i]=-INF;
f[x][b[x]]=g[x][b[x]]=;
for (int i=head[x]; i; i=edge[i].next)
if (edge[i].to!=fa)
{
int to=edge[i].to;
Dfs(to,x);
for (int j=; j<=size[x]+size[to]; ++j)
tmpf[j]=INF,tmpg[j]=-INF;
for (int j=; j<=size[x]; ++j)
for (int k=; k<=size[to]; ++k)
{
tmpf[j+k]=min((int)tmpf[j+k],(int)f[x][j]+f[to][k]);
tmpg[j+k]=max((int)tmpg[j+k],(int)g[x][j]+g[to][k]);
}
size[x]+=size[to];
for (int j=; j<=size[x]; ++j)
{
f[x][j]=min(f[x][j],tmpf[j]);
g[x][j]=max(g[x][j],tmpg[j]);
}
}
for (int i=; i<=size[x]; ++i)
if (f[x][i]<INF)
ans[f[x][i]][i]++,ans[g[x][i]+][i]--;
} int main()
{
scanf("%d",&T);
while (T--)
{
memset(ans,,sizeof(ans));
memset(head,,sizeof(head));
num_edge=;
scanf("%d%d",&n,&q);
for (int i=; i<=n-; ++i)
scanf("%d%d",&u,&v),add(u,v),add(v,u);
for (int i=; i<=n; ++i)
scanf("%d",&b[i]);
Dfs(,);
for (int i=; i<=n; ++i)
for (int j=; j<=n; ++j)
ans[i][j]+=ans[i-][j];
for (int i=; i<=q; ++i)
{
scanf("%d%d",&u,&v);
if (ans[u][v]) puts("YES");
else puts("NO");
}
puts("");
}
}