HDU 4707 Pet(DFS：树深度相关)

HDU 4707 Pet(DFS：树深度相关)

http://acm.hdu.edu.cn/showproblem.php?pid=4707

题目描述：

        给出一个具有N（0<N<=100000）个节点的树（节点编号从0到N-1），然后给出一个距离D(0<D<N),求出距离0号节点的距离大于D的节点个数。树中相邻节点的距离为1。

输入：首先是一个T（0<T<=10），表实例个数。对于每个实例，首先是一行N和D，然后是N-1行（X,Y），表示树的每条边。

输出：输出距离0号节点距离大于D的节点总数。

分析：

        从0号节点开始做一次DFS，记录距离大于D的节点数即可。本题采用的就是不走连接父节点边的策略判重。否则就需要一个vis数组来判重。

AC代码：

#include<cstdio>
#include<vector>
using namespace std;
const int maxn=100000+100;
vector<int> G[maxn];
int sum,n,d;
void dfs(int i,int dep,int fa)
{
    if(dep>d) sum++;
    int m=G[i].size();
    for(int j=0;j<m;j++)if(fa!=G[i][j])
        dfs(G[i][j],dep+1,i);
}
int main()
{
    int t;
    scanf("%d",&t);
    while(t--)
    {
        sum=0;
        scanf("%d %d",&n,&d);
        for(int i=0;i<n;i++)
            G[i].clear();
        for(int i=0;i<n-1;i++)
        {
            int x,y;
            scanf("%d%d",&x,&y);
            G[x].push_back(y);
            G[y].push_back(x);
        }
        dfs(0,0,-1);
        printf("%d\n",sum);
    }
    return 0;
}