（06-08补）UVALive 3902 Network 网络

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

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

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

复杂度O （N）

本代码存在部分bug：实在未能发现，望指导

#include <cstdio>
#include <cstring>
#include <vector>
#include <iostream>
#include <algorithm>

using namespace std;

const int maxn = 1005;
vector <int> node[maxn], mapp[maxn];
int fa[maxn], n, s, k;
bool vis[maxn];

void dfs(int u, int f, int d){ //寻找叶子结点
fa[u] = f;
if (mapp[u].size() == 1 && d>k) node[d].push_back(u); //避开按深度排序
for (int i = 0; i<mapp[u].size(); i++){
int v = mapp[u][i];
if (v != f) dfs(v, u, d+1);
}
}

void dfs2(int u, int f, int d){
vis[u] = true;
for (int i = 0; i<mapp[u].size(); i++){
int v = mapp[u][i];
if (v != f && d<k) dfs(v, u, d+1);
}
}

int solve(){
int ans = 0;
memset(vis, 0, sizeof(vis));
for (int d = n-1; d>k; d++)
for (int i = 0; i<node[d].size(); i++){ //从深度最大出发，贪心
int u = node[d][i];
if (vis[u]) continue;

int v = u;
for (int i = 0; i<k; i++) v = fa[v]; //寻找k级祖先
dfs2(v, -1, 0);
ans++;
}
return ans;
}

int main(){
int T;
scanf("%d", &T);
while (T--){
scanf("%d%d%d", &n, &s, &k);
for (int i = 1; i<=n; i++){
mapp[i].clear();
node[i].clear();
}
for (int i = 0; i<n; i++){
int a, b;
scanf("%d%d", &a, &b);
mapp[a].push_back(b);
mapp[b].push_back(a);
}
dfs(s, -1, 0);
printf("%d\n", solve());
}
return 0;
}