UVaLive（LA） 3902 Network，Seoul 2007

题目传送门

---

显然先让第一个服务器成为根节点，然后我们搜一遍，先把所有叶节点入到一vector里面，vector的第一维是深度，出来之后让最深的叶节点晚上找第k级fa，因为这样可以让其影响力最大，然后去往所有方向搜这个fa能影响到的地方，然后打上标记，出来之后继续for叶节点，直到所有的叶节点都被覆盖位置。

#include<cstdio>
#include<cstring>
#include<iostream>
#include<vector>
using namespace std;
const int MAXN=1000+10;
int T,n,s,k,ff,tt,fa[MAXN];
vector<int>son[MAXN],nodes[MAXN]; 
bool covered[MAXN];
void dfs(int u,int father,int dep){
    fa[u]=father;
    int nc=son[u].size();
    if(nc==1&&dep>k) nodes[dep].push_back(u);
    for(register int i=0;i<nc;i++){
        int v=son[u][i];
        if(v==father) continue;
        dfs(v,u,dep+1);
    }
}
void dfs2(int u,int fa,int dep){
    covered[u]=true;
    int nc=son[u].size();
    for(register int i=0;i<nc;i++){
        int v=son[u][i];
        if(v!=fa&&dep<k) dfs2(v,u,dep+1);
    }
}
int solve(){
    int ans=0;
    memset(covered,0,sizeof(covered));
    for(register int d=n-1;d>k;d--){
        for(register int i=0;i<nodes[d].size();i++){
            int u=nodes[d][i];
            if(covered[u]) continue;
            int v=u;
            for(register int j=0;j<k;j++) v=fa[v];
            dfs2(v,-1,0);
            ans++;
        }
    } 
    return ans;
}
int main(){
    scanf("%d",&T);
    for(register int i=1;i<=T;i++){
        scanf("%d%d%d",&n,&s,&k); 
        for(register int i=1;i<=n;i++){son[i].clear();nodes[i].clear();}
        for(register int i=1;i<=n-1;i++){
            scanf("%d%d",&ff,&tt);
            son[ff].push_back(tt);son[tt].push_back(ff);
        }
        dfs(s,-1,0);
        printf("%d\n",solve());
    }
    return 0;
}