UVA 1267 Network(dfs)

先把s节点作为根节点，建立一棵树，k层以上是不用管的，把k层以下的叶子结点放到node数组中，最下面的叶子结点网上走k-1层，然后当前节点作为根，建树，重复循环直到所有的点都被覆盖为止。

////  main.cpp//  Richard////  Created by 邵金杰 on 16/9/30.//  Mail:[e-mail] or [e-mail]//  Copyright © 2016年 邵金杰. All rights reserved.//#include<iostream>#include<cstdio>#include<cstring>#include<algorithm>#include<vector>using namespace std;const int maxn=1000+10;vector<vector<int> > G(maxn),node(maxn);int n,s,k;int fa[maxn];int covered[maxn];void dfs1(int u,int f,int d){    fa[u]=f;    int nc=(int)G[u].size();    if(nc==1&&d>k) node[d].push_back(u);    for(int i=0;i<G[u].size();i++){        int v=G[u][i];        if(v!=f) dfs1(v,u,d+1);    }}void dfs2(int u,int f,int d){    covered[u]=1;    for(int i=0;i<G[u].size();i++){        int v=G[u][i];        if(v!=f&&d<k) dfs2(v,u,d+1);    }}int solve(){    int ans=0;    memset(covered,0,sizeof(covered));    for(int d=n-1;d>k;d--)    {        for(int i=0;i<node[d].size();i++){            int p=node[d][i];            if(covered[p]) continue;            int v=p;            for(int j=0;j<k;j++) v=fa[v];            dfs2(v,-1,0);            ans++;        }    }    return ans;}int main(){    int t;    scanf("%d",&t);    while(t--)    {        int u,v;        scanf("%d%d%d",&n,&s,&k);        for(int i=0;i<=n;i++) G[i].clear(),node[i].clear();        for(int i=0;i<n-1;i++)        {            scanf("%d%d",&u,&v);            G[u].push_back(v);            G[v].push_back(u);        }        dfs1(s,-1,0);        printf("%d\n",solve());    }    return 0;}//2 14//12 2//1 2//2 3//3 4//4 5//5 6//7 5//8 5//4 9//10 3//2 12//12 14//13 14//14 11