（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;
}