UVALive - 3902 Network 无根树转为有跟树+搜索+贪心

http://acm.hust.edu.cn/vjudge/problem/viewProblem.action?id=16451

题目大意：n台机器连成一个树状网络，其中叶节点是客户端，其他节点是服务器。现在有一台服务器在节点s，服务器能传播的信号的距离为k，因为有的用户距离服务器的距离大于k，所以必须添加服务器。问最少要添加几个服务器，才能使每个客户端都收到信号

首先以S为根节点建立一颗有根树，并dfs建树时记录点的深度，是否为叶子。

把叶子按深度从大到小排序，每次贪心选一个深度最大的叶子，然后找到他的k级祖先，以k级祖先出发dfs标记K层以内的叶子，如此反复直到叶子都被染色。

复杂度O （N）

#include <cstdio>
#include <cmath>
#include <cstring>
#include <string>
#include <algorithm>
#include <queue>
#include <map>
#include <set>
#include <vector>
#include <iostream>
using namespace std;

const double pi=acos(-1.0);
double eps=0.000001;

int n;
vector< int >mp[1005];
struct node
{
int x,v;
node(int a=0,int b=0)
{
x=a,v=b;
}
};
bool cmp(node a,node b)
{
return a.v>b.v;
}
node leave[1005];
int s,k;
int id=0;
int fa[1005];
int vis[1005];
int isleaf[1005];
void dfs1(int x,int dep)
{
vis[x]=1;
if (mp[x].size()==1)
{
leave[++id]=node(x,dep);
isleaf[x]=1;
return ;
}
for (int i=0; i<mp[x].size(); i++)
{
int v=mp[x][i];
if (vis[v])continue;
fa[v]=x;
dfs1(v,dep+1);
}
}
int find_fa(int x )
{

int tmp=k;
while(tmp--)
x=fa[x];
return x;
}
void dfs2(int x,int cur)
{
if (isleaf[x])
vis[x]=1;
for (int i=0;i<mp[x].size();i++)
{
int v=mp[x][i];
if (vis[v])continue;
if (cur>0)
dfs2(v,cur-1);
}

}

int main()
{

int t;
cin>>t;
int i,j;
int x,y;
while(t--)
{
id=0;
cin>>n;
cin>>s>>k;
for (i=1; i<=n; i++) mp[i].clear();
for (i=1; i<=n-1; i++)
{
scanf("%d%d",&x,&y);
mp[x].push_back(y);
mp[y].push_back(x);
}
memset(vis,0,sizeof vis);
memset(isleaf,0,sizeof isleaf);
memset(fa,0,sizeof fa);
dfs1(s,1);
sort(leave+1,leave+1+id,cmp);
memset(vis,0,sizeof vis);
int ans=0;
dfs2(s,k);

for (i=1;i<=id;i++)
{
node tp=leave[i];
if (vis[tp.x])continue;
int st=find_fa(tp.x );
ans++;
dfs2(st,k);
}

printf("%d\n",ans);
}

return 0;

}