Tree UVALive - 8212

Posted 2021-08-01 Jozky86

Tree UVALive - 8212

题意：

有n个点，k个颜色，每个点都要被染色，相同颜色之间的边算是被该颜色覆盖，问有多少边被所有颜色覆盖

题解：

题目给的是无根树，我们可以将1默认为根然后求所有点的子树大小，
我们可以这么考虑，对于图中的红边，红边两侧如果点的数量大于等于K的数量，也就是说所有颜色均可以在红边两侧，说明红边可以被所有颜色覆盖，贡献为1。
如何知道红边两侧数量？就是求子树大小，一侧为子树的大小，另一侧为n-子树大小

代码：

#include <bits/stdc++.h>
using namespace std;
#define ll long long
#define INF 0x3f3f3f3f
const int maxn=2e5+9;
vector<int>vec[maxn];
int siz[maxn];
int getsiz(int u,int fa){
	for(int i=0;i<vec[u].size();i++){
		int v=vec[u][i];
		if(v==fa)continue;
		siz[v]=1;
 		siz[u]+=getsiz(v,u);
	}
	return siz[u];
}
int main(){
	int t;
	scanf("%d",&t);
	while(t--){
		memset(siz,0,sizeof(siz));
		int n,k;
		scanf("%d%d",&n,&k);
		for(int i=1;i<n;i++){
			int a,b;
			scanf("%d%d",&a,&b);
			vec[a].push_back(b);
			vec[b].push_back(a);
		}
		siz[1]=getsiz(1,0)+1;
		int tot=0;
		for(int i=1;i<=n;i++){
			int x,y;
			x=siz[i];
			y=n-x;
			if(x>=k&&y>=k)tot++;
		}
		printf("%d\\n",tot);
		for(int i=1;i<=n;i++){
			vec[i].clear();
		}
	}
	return 0;
}