CF1593 E. Gardener and Tree （树上删叶子，BFS）

E. Gardener and Tree

分析：

• 在树上，从外层到里层逐层删叶子

  跑一遍 $BFS$ 即可

#include <bits/stdc++.h>
using namespace std;

const int N=1e6+5;
vector <int> g[N];
int du[N],dep[N];
int sub;
void bfs(int n,int k)
{
	queue <int> q;
	for(int i=1;i<=n;i++) 
	{
		if(du[i]==1) 
		{
			q.push(i);
			sub++;
			du[i]=0; dep[i]=1; 
		}
	}
	while(!q.empty())
	{
		int u=q.front();
		q.pop();
		for(auto v : g[u])
		{
			du[v]--;
			if(du[v]==1 && dep[u]<k) //根据du判断是否是叶子，dep判断处于第几层
			{
				q.push(v);
				sub++;
				dep[v]=dep[u]+1;
			}
		}
	} 
}
signed main()
{
	int T;
	scanf("%d",&T);
	while(T--)
	{
		int n,k;
		scanf("%d%d",&n,&k);
		for(int i=1;i<=n;i++) du[i]=0, dep[i]=1, g[i].clear();
		sub=0;
		for(int i=1;i<n;i++) 
		{
			int u,v;
			scanf("%d%d",&u,&v);
			g[u].push_back(v);
			g[v].push_back(u);
			du[u]++; du[v]++;
		}
		if(n==1 || 2*k>=n) { puts("0"); continue; } // 最长就是一条链的情况，从外到里删掉k层就删完了
		bfs(n,k);
		printf("%d\n",n-sub);
	}
}