HDU - 6228 Tree —— 思维

题意：

给一棵树的每个节点染色，有k个颜色可以选，染成相同颜色的节点的路径也会形成一个树，这样每种颜色就都会有一棵树，问这些树的交集最多有多少条边。

思路：

开始想的时候是从叶子节点一圈一圈向里染，每个节点形成的子树的节点个数如果大于等于k，那这个节点上面的所有节点就都可以算进交集里，但是搜索的时候T了。

后来想对于每条边，如果这条边左侧右侧都有大于等于k个节点，那么这条边就一定能被算进交集中。

dfs一下，计算子树节点个数。

#include <bits/stdc++.h>
using namespace std;
int n,k;
vector<int>v[200010];
bool vis[200010];
int num[200010];
int dfs(int x)
{
	int sum=1;
	for(int i=v[x].size()-1;i>=0;i--)
	{
		if(vis[v[x][i]]==false)
		{
			vis[v[x][i]]=true;
			sum+=dfs(v[x][i]);
		}
	}
	num[x]=sum;
	return sum;
}
int main()
{
	int t;
	scanf("%d",&t);
	while(t--)
	{
		scanf("%d%d",&n,&k);
		memset(vis,false,sizeof vis);
		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);
		}
		vis[1]=true;
		dfs(1);
		int ans=0;
		for(int i=1;i<=n;i++)
		{
			if(num[i]>=k&&n-num[i]>=k)
			ans++;
		}
		printf("%d\n",ans);
	}
}