【BZOJ 5072】小A的树

题目描述

小 A 成为了一个园艺家！他有一棵 $n$ 个节点的树。他不小心打翻了墨水瓶，使得树的一些节点被染黑了。小 A 发现这棵染黑了的树很漂亮，于是想从树中取出一个 $x$ 个点的联通子图，使得这些点中恰有 $y$ 个黑点，他想知道他的愿望能否实现。可是他太小，不会算，请你帮帮他。
$T\le 5$，$n\le 5000$，$q\le 10^5$，$1\le u_i,v_i\le n$， b i ∈ { 0 , 1 } b_i\in\{0,1\} bi​∈{0,1}，$1\le x_i\le n$，$0\le y_i\le n$，$y_i\le x_i$。

算法分析

有一个结论：同一棵树中 $x$ 个点的连通子图的最小黑点数和最大黑点数之间的答案都能满足，因为我们可以不断往一个图里加点，而每次最多加 $1$ 个黑点。

设 $f[i][j]$ 为以 $i$ 为根的子树中包含点 $i$ 且大小为 $j$ 的连通子图中的最小黑点个数，$g[i][j]$ 为其最大黑点个数，利用树形背包 DP 可以很方便地求出两者的值（代码中的枚举方法效率更高，以后更新一下代码风格），则大小为 $x$ 的连通子图包含的黑点数最小为 m i n { f [ i ] [ x ] } min\{f[i][x]\} min{f[i][x]}，最大为 m a x { f [ i ] [ x ] } max\{f[i][x]\} max{f[i][x]}，对于每组询问，判断 $y$ 是否在两者之间即可。

注意初始化和输出格式。

代码实现

#include <cstdio>
#include <cstring>
#include <algorithm>
#define cmin(x,y) x=std::min(x,y)
#define cmax(x,y) x=std::max(x,y)
const int maxn=5005;
int head[maxn],ev[maxn<<1],nxt[maxn<<1],idx=0;
inline void add(int u,int v) {ev[++idx]=v;nxt[idx]=head[u];head[u]=idx;}
int n,b[maxn],tot[maxn],f[maxn][maxn],g[maxn][maxn],ansf[maxn],ansg[maxn];
void dfs(int x,int fa) {
	memset(f[x],0x3f,sizeof(f[x]));memset(g[x],0xc0,sizeof(g[x]));
	f[x][1]=g[x][1]=(bool)b[x];tot[x]=1;
	for(int i=head[x];i;i=nxt[i]) {
		int v=ev[i];if(v==fa) continue;
		dfs(v,x);
		for(int j=tot[x];j>=1;--j)
			for(int k=tot[v];k>=1;--k) {
				cmin(f[x][j+k],f[x][j]+f[v][k]);cmax(g[x][j+k],g[x][j]+g[v][k]);
			}
		tot[x]+=tot[v];
	}
	for(int i=1;i<=n;++i) {cmin(ansf[i],f[x][i]);cmax(ansg[i],g[x][i]);}
}
int main() {
	int t;scanf("%d",&t);
	while(t--) {
		int q;scanf("%d%d",&n,&q);
		int m=n-1,u,v;memset(head,0,sizeof(head));idx=0;
		while(m--) {
			scanf("%d%d",&u,&v);
			add(u,v);add(v,u);
		}
		for(int i=1;i<=n;++i) scanf("%d",&b[i]);
		memset(ansf,0x3f,sizeof(ansf));memset(ansg,0xc0,sizeof(ansg));
		dfs(1,1);
		while(q--) {
			int x,y;scanf("%d%d",&x,&y);
			printf("%s\n",(ansf[x]<=y&&y<=ansg[x])?"YES":"NO");
		}
		if(t) printf("\n");
	}
	return 0;
}