hdu:color a tree-树形dp

我们可以二分答案。(至少染色几个点)

然后对于限制B,我们可以转化成子树x内之多有num(总染色数)-x(子树外的染色数)

然后通过树形dp从下到上算每个根节点的子树内的上界和下界,判断上是否不小于下界。

PS:还有注意一下在算上届的时候应该是所有子树的up和+1(因为还有根节点自己)->被坑了我好难受。

这样也可以保证对于没有上届限制的点,他的上届就是siz

同时还要注意判断是否up,down的值符合实际(即考虑>=0和<=siz)

#include <iostream>
#include <vector>
#include <cstring>
#include <algorithm>
#include <cmath>
#include <cstdio>

using namespace std;
const int maxn=100005;
int n;
vector<int> v[maxn]; 
int up[maxn],down[maxn],f[maxn],na[maxn],nb[maxn],sz[maxn];
int find(int num,int x,int fa)
{
	up[x]=1;down[x]=0;
	for(int i=0;i<v[x].size();i++)
	{
		if(v[x][i]==fa) continue;
		if(!find(num,v[x][i],x))
		{
			return 0; 
		}
		up[x]+=up[v[x][i]];
		down[x]+=down[v[x][i]];
	} 
	up[x]=min(up[x],min(sz[x],num-nb[x]));
	down[x]=max(down[x],na[x]); 
	return down[x]<=up[x]&&down[x]<=sz[x]&&up[x]>=0;
}
int check(int x)
{
	if(find(x,1,0)&&down[1]<=x&&x<=up[1]) return 1;
	return 0;
}
void getsize(int x,int fa)
{
	sz[x]=1;
	for(int i=0;i<v[x].size();i++)
	{
		if(v[x][i]==fa) continue;
		getsize(v[x][i],x);
		sz[x]+=sz[v[x][i]];
	}
}
void work()
{
	memset(na,0,sizeof(na));
	memset(nb,0,sizeof(nb));
	scanf("%d",&n);
	for(int i=1;i<=n;i++) v[i].clear();
	for(int i=1;i<n;i++)
	{
		int x,y;scanf("%d%d",&x,&y);
		v[x].push_back(y); v[y].push_back(x);
	}
	getsize(1,0); //
	int a,b;scanf("%d",&a);int flag=1;
	for(int i=1;i<=a;i++)
	{
		int x,y;scanf("%d",&x);scanf("%d",&y);
		na[x]=max(y,na[x]); 
		if(na[x]>sz[x]) {flag=0;}	//
	}
	scanf("%d",&b);
	for(int i=1;i<=b;i++) 
	{
		int x,y;scanf("%d",&x);
		scanf("%d",&y);	
		nb[x]=max(nb[x],y);
		if(nb[x]>n-sz[x]) flag=0; //
	}
	if(!flag) {printf("-1\n");return;}
	int l=0,r=n;int ans=-1;
	while(l<=r)
	{
		int mid=(l+r)/2;
//		cout<<mid<<" "<<check(mid)<<endl;
	//	cout<<"mid="<<mid<<endl;
		if(check(mid))
		{
			ans=mid;
			r=mid-1;
		}
		else l=mid+1;
	}
	printf("%d\n",ans);
}
int main()
{
	int T;scanf("%d",&T);
	for(int ti=1;ti<=T;ti++) work();
	return 0; 
}

 

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值