bzoj5072 小A的树 题解

题意

给出一棵 n 个点的树，每个点有黑白两种颜色。q 次询问，每次
询问给出 x 和 y，问能否选出一个 x 个点的联通子图，使得其中
黑点数目为 y。

范围

n ≤ 5000,q ≤ 10^5

其实证明我也不会没弄懂，只是听老师讲了，我们可以猜想：对于某一大小的连通子图，其包含黑点数的最小值与最大值之间的所有点数目都能够取得到。

证明：证明很简单，考虑从最小值一个个删除点并加入点到最大值的过
程，黑点个数每次最多变化 1，因此能遍历从最小值到最大值中
的所有点。（from dzy）

我们定义f[x][y]表示以x为根的树中有y个黑点的最小节点数

同理g[x][y]表示以x为根的树中有y个黑点的最大节点数

然后这题便可以树上背包解决，时间复杂度（n^2）

代码

#include<bits/stdc++.h>
using namespace std;
const int Max=5010;
int n,T,q,tot,u,v,root,xi,yi;
int ver[Max*2],head[Max],Next[Max*2];
int score[Max];
int f[Max][Max],g[Max][Max],siz[Max],ff[Max],gg[Max];
void add(int x,int y){ 
	ver[++tot]=y;Next[tot]=head[x];head[x]=tot;
}
void dp(int x,int fa){//由于存树时用的是双向图，此处要判断 
	siz[x]=1;
	g[x][score[x]]=1;//初始化保证g[x][0] or f[x][0]为1，否则最小值永远是0 
	f[x][score[x]]=1;
	for(int i=head[x];i;i=Next[i]){
		int y=ver[i];
		if(fa==y) continue;
		dp(y,x);
		memcpy(ff,f[x],sizeof f[x]);//由于在进行背包的过程中求得的不一定是最优解，故用临时数组进行储存 
		memcpy(gg,g[x],sizeof g[x]);
		for(int t=siz[x];t>=score[x];--t){//两棵树的合并 
			for(int j=siz[y];j>=score[y];--j){
				ff[t+j]=min(ff[t+j],f[x][t]+f[y][j]);
				gg[t+j]=max(gg[t+j],g[x][t]+g[y][j]);
			}
		}
		siz[x]+=siz[y];
		for(int j=score[x];j<=siz[x];++j){
			f[x][j]=ff[j];
			g[x][j]=gg[j];
		}	
	}
	for(int i=0;i<=siz[x];++i){//用g[0][x] f[0][x]储存答案 
		f[0][i]=min(f[0][i],f[x][i]);
		g[0][i]=max(g[0][i],g[x][i]);
	}
	return;
}
int main(){
	//freopen("trees.in","r",stdin);
	//freopen("trees.out","w",stdout);
	scanf("%d",&T);
	while(T--){
		scanf("%d %d",&n,&q);
		memset(head,0,sizeof(head));
		memset(f,0x3f,sizeof(f));
		memset(g,0xcf,sizeof(g));
		tot=0;
		for(int i=1;i<n;++i){
			scanf("%d %d",&u,&v);
			add(u,v);
			add(v,u);
		}
		for(int i=1;i<=n;++i){
			scanf("%d",&score[i]);
		}
		dp(1,0);
		while(q--){
			scanf("%d %d",&xi,&yi);
			if(xi>=f[0][yi]&&xi<=g[0][yi]){
				puts("YES");
			}
			else puts("NO");
		}printf("\n");
	}
	return 0;
}

注：bzoj需加快读才能过，比较卡时间