[unknown OJ] 小A的树

最新推荐文章于 2021-07-16 17:39:30 发布

C202044zxy

最新推荐文章于 2021-07-16 17:39:30 发布

阅读量160

点赞数

分类专栏：树形dp

本文链接：https://blog.csdn.net/C202044zxy/article/details/109999248

版权

树形dp 专栏收录该内容

33 篇文章 1 订阅

订阅专栏

一、题目

点此看题

二、解法

明显的定义是 $d p [u] [x] [y]$ 表示点 $u$ 子树内选了 $x$ 个点， $y$ 个是黑点的联通子图能不能达到，但是状态就爆了，显然不行。

需要猜结论：如果 $x$ 个点的最小黑点数是 $l$ ，最大黑点数是 $r$ ，那么黑点 $[l, r]$ 的联通子图都存在。证明：我们增加黑点的方式是增加一个点在删一个点，此时黑点变化数最多为 $1$ ，所以黑点函数可以视为在整点处连续变化，所以 $[l, r]$ 个数的黑点一定取得到，其他的一定取不到。

然后就是一个树形背包板子，设 $d p [u] [i] [0 / 1]$ 为 $u$ 子树大小为 $i$ 的黑点最小 $/$ 最大值，然后跑树形背包就行了，时间复杂度是 $O(n^2)$ 的，因为点对的影响只会在 $\tt lca$ 处被计算一次。

处理出 $d p$ 数组后就直接差分了（考场上我竟然没想到 [惊恐] ）

#include <cstdio>
#include <iostream>
using namespace std;
const int M = 5001;
const int inf = 0x3f3f3f3f;
#define rint register int
int read()
{
	int x=0,f=1;char c;
	while((c=getchar())<'0' || c>'9') {if(c=='-') f=-1;}
	while(c>='0' && c<='9') {x=(x<<3)+(x<<1)+(c^48);c=getchar();}
	return x*f;
}
int T,n,m,tot,f[M],a[M],siz[M],f1[M][M],f2[M][M],b[M];bool c[M][M];
struct edge
{
	int v,next;
	edge(int V=0,int N=0) : v(V) , next(N) {}
}e[2*M];
int Min(int x,int y)
{
	if(x<y) return x;
	return y;
}
int Max(int x,int y)
{
	if(x>y) return x;
	return y;
}
void dfs(int u,int fa)
{
	siz[u]=1;
	if(a[u]==1) f1[u][1]=f2[u][1]=1;
	else f1[u][1]=f2[u][1]=0;
	for(int t=f[u];t;t=e[t].next)
	{
		int v=e[t].v;
		if(v==fa) continue;
		dfs(v,u);
		for(rint i=siz[u];i>=0;i--)
			for(rint j=siz[v];j>=0;j--)
			{
				f1[u][i+j]=Min(f1[u][i+j],f1[u][i]+f1[v][j]);
				f2[u][i+j]=Max(f2[u][i+j],f2[u][i]+f2[v][j]);
			}
		siz[u]+=siz[v];
	}
}
int main()
{
	T=read();
	while(T--)
	{
		n=read();m=read();
		for(int i=1;i<=n;i++)
			for(int j=0;j<=n;j++)
			{
				f1[i][j]=inf;
				f2[i][j]=-inf;//都要清空 
			}
		for(int i=1;i<n;i++)
		{
			int u=read(),v=read();
			e[++tot]=edge(v,f[u]),f[u]=tot;
			e[++tot]=edge(u,f[v]),f[v]=tot; 
		}
		for(int i=1;i<=n;i++)
			a[i]=read();
		dfs(1,0);
		for(rint i=1;i<=n;i++)//枚举siz 
		{
			for(rint j=0;j<=i;j++) b[j]=0;
			for(rint j=1;j<=n;j++)//枚举点 
				if(f1[j][i]<=f2[j][i])
				{
					b[f1[j][i]]++;
					b[f2[j][i]+1]--; 
				}
			for(rint j=0;j<=i;j++)//枚举black points 
			{
				if(j>0) b[j]+=b[j-1];
				c[i][j]=(b[j]>0);
			}
		}
		for(int i=1;i<=m;i++)
		{
			int x=read(),y=read();
			if(c[x][y]) puts("YES");
			else puts("NO");
		}
	}
}
/*
感觉是个背包问题 
30%:n<=1000,q<=1000
100%:n<=5000
*/