Codeforces Round #656 (Div. 3) F. Removing Leaves（补题）

题意：有一棵树，共有n个节点。每次从同一个节点上去除k个叶子，问最多去除几次。

解法：这题有些可惜了，看到做出来的人少觉得不能做，实际上不难。就是一个贪心。贪心的策略是：每次剪掉即可。因为剪去叶子不会对其他的节点产生负影响。也就是说，剪除一次以后，只会说多出一个节点可以减，不会说剪除了以后导致其他节点减不了。因此这道题根本不用考虑剪枝的先后问题，直接剪就行了。

具体做法：维护每个点的出入度，可以叫做“关联度”。当每有一个点相连的时候，关联度加一。当关联度为1的时候就是叶子节点，也就是马上要被剪除的节点。然后再维护一个叶子数。每一个叶子找到它的爸爸然后给爸爸加一个叶子数。

然后开一个队列，把叶子数大于等于k的全部放入到队列里面。然后开始剪除，令tmp=数量/k，ans+=tmp。叶子数-=tmp*k，关联度-=tmp*k。如果此时该点已经变成一个叶子节点（即叶子数==0并且关联度==1）则遍历该点连接的点，若连接的是有儿子的点（叶子数！=0），则这个点的叶子数++。如果此时上面的这个点的叶子数正好等于k，那么就放入（如果原来就大于等于k应该已经在里面了）。

执行以上操作直到队列清空。

 

代码：

#include<bits/stdc++.h>
using namespace std;
const int N=200005;
int n,k;
vector<int> ed[N];//邻接矩阵
int guan[N];//关联度
int ye[N];//叶子数 
int qu[N];//表示被去除的叶子 
queue<int> q;
int ans; 
void jjwt()
{
	ans=0;
	scanf("%d%d",&n,&k);
	for (int i=0;i<n;i++)
		ed[i].clear();
	memset(guan,0,sizeof(guan));
	memset(ye,0,sizeof(ye));
	for (int i=0;i<n-1;i++)
	{
		int x,y;
		scanf("%d%d",&x,&y);
		x--;y--;
		ed[x].push_back(y);ed[y].push_back(x);
		guan[x]++;guan[y]++;
	}
	if (k==1)
	{
		printf("%d\n",n-1);
		return;
	}
	for (int i=0;i<n;i++)
	{
		//当某个点只有一个关联度的时候，说明它是叶子，
		//就让它连接的那个有了一个叶子节点 
		if (guan[i]==1)
		{
			ye[ed[i][0]]++;
		}
	}
	for (int i=0;i<n;i++)
	{
		if (ye[i]>=k)
		{
			q.push(i);
		}
	}
	while (!q.empty())
	{
		int v=q.front();
		q.pop();
		//对每一个在队列的节点进行剪除操作 
		int tmp=ye[v]/k;
		ye[v]-=tmp*k;//减去tmp*k个叶子
		guan[v]-=tmp*k;
		ans+=tmp;
		if (ye[v]==0&&guan[v]==1)//如果已经成为了叶子
		{
			guan[v]=0;
			for (int i=0;i<ed[v].size();i++)
			{
				//这里不能用有没有叶子节点来判断，而要用关联节点来判断，因为父节点不一定有叶子节点 
				if (guan[ed[v][i]]>1)//连接的父亲节点 
				{
					ye[ed[v][i]]++;//给父亲节点增加一个叶子
					if (ye[ed[v][i]]==k)
						q.push(ed[v][i]);
				} 
			}
		} 
		
	}
	printf("%d\n",ans);
}
int main()
{
	int T;
	scanf("%d",&T);
	while (T--)
	{
		jjwt();
	}
	return 0;
}

反思：本题其实不难，只要想到按题意模拟，前后剪枝不会影响结果这一点，问题就迎刃而解了。但是注意新增叶子节点的父节点应该如何判断比较麻烦。一开始用的是有其他叶子，wa5.后来发现父亲节点不一定有其他叶子，用有>1个关联节点就可以很好的解决问题。

另外，比较难想的是叶子节点还要不要处理了。因为叶子节点的关联度为1，不可能进入队列以及循环，因此直接舍弃即可。我们发现进入队列一定要叶子大于等于k，进入循环也必须要有关联度>1，因此旧的叶子去了也就去除了，不会再对后面的操作有影响了。