Live Archive 3902 Network

题意:n台机器连成一个网络，其中叶子节点为客户端，其他节点为服务器，初始时只有一个服务器装载了程序，但是这个程序只能让离该服务器的距离不大于k的客户端使用，现在你可以选出一些服务器安装上这种程序，请你求出使所有客户端能使用程序的最小安装次数

树上的贪心，我们可以将给出的那一个点当做树根然后建树。

接下来我们从深度最大的点讨论，很容易发现如果u号点是客户端且没有被覆盖，那么最优的计划就是从u号点向上走k步到达的祖先上安装程序最优，因为这样可以尽量使得更上面的叶子节点使用到程序

#include<cstdio>
#include<iostream>
#include<cstring>
#include<vector>
using namespace std;
const int maxn=1005,inf=1e9;
inline void _read(int &x){
    char t=getchar();bool sign=true;
    while(t<'0'||t>'9')
    {if(t=='-')sign=false;t=getchar();}
    for(x=0;t>='0'&&t<='9';t=getchar())x=x*10+t-'0';
    if(!sign)x=-x;
}
int t,n,last[maxn],k,p,ans,fa[maxn],d[maxn];
bool Cover[maxn];
struct Edge{
	int a,b,Next;
	Edge(int a,int b,int Next):a(a),b(b),Next(Next){}
};
vector<Edge>s;
void insert(int a,int b){
	s.push_back(Edge(a,b,last[a]));
	last[a]=s.size()-1;
	s.push_back(Edge(b,a,last[b]));
	last[b]=s.size()-1;
}
vector<int>leaf[maxn];
void dfs1(int u,int dep,int f){
	fa[u]=f;
	if(d[u]==1&&dep>k){
		leaf[dep].push_back(u);
	}
	int i,v;
	for(i=last[u];i>=0;i=s[i].Next){
		v=s[i].b;
		if(v==f)continue;
		dfs1(v,dep+1,u);
	}
}
void dfs2(int u,int dep,int f){
	Cover[u]=1;
	for(int i=last[u];i>=0;i=s[i].Next){
		int v=s[i].b;
		if(v==f)continue;
		if(dep<k)dfs2(v,dep+1,u);
	}
}
void solve(){
	int i,j,u,v,dep;
	for(dep=n-1;dep>k;dep--)
	    for(i=0;i<leaf[dep].size();i++){
	    	v=leaf[dep][i];
	    	if(Cover[v])continue;
	    	int u=v;
	    	for(j=0;j<k;j++)u=fa[u];
	    	dfs2(u,0,0);
	    	ans++;
		}
}
void clear(){
	s.clear();
	memset(last,-1,sizeof(last));
	memset(d,0,sizeof(d));
	ans=0;
	memset(Cover,0,sizeof(Cover));
}
int main(){
	_read(t);
	while(t--){
		clear();
		_read(n);_read(p);_read(k);
		int i,x,y;
		for(i=1;i<n;i++)leaf[i].clear();
		for(i=1;i<n;i++){
			_read(x);_read(y);
			insert(x,y);
			d[x]++,d[y]++;
		}
		dfs1(p,0,0);
		solve();
		printf("%d\n",ans);
	}
}