hdu - 4707 - Pet

最新推荐文章于 2020-12-23 06:31:24 发布

原创最新推荐文章于 2020-12-23 06:31:24 发布 · 1.5k 阅读

0 ·

CC 4.0 BY-SA版权

water 专栏收录该内容

26 篇文章

订阅专栏

本文介绍了如何在一个N个结点的树形结构中，统计那些距离根节点的距离大于给定值D的结点数量。涉及到的数据范围包括测试组数T、结点数量N以及距离阈值D。通过DFS深度优先搜索算法来解决该问题。

题意：一棵N个结点(编号从0开始)的树，根结点为0，求到根结点的距离大于D的结点个数（0 < 测试组数T <= 10, 0<N<=100000, 0<D<N）。

题目链接：http://acm.hdu.edu.cn/showproblem.php?pid=4707

——>>统计吧。。。

#include <cstdio>
#include <cstring>

using namespace std;

const int maxn = 100000 + 10;

int D, head[maxn], nxt[maxn<<1], v[maxn<<1], ecnt, d[maxn], ret;

void init(){
    memset(head, -1, sizeof(head));
    ecnt = 0;
    ret = 0;
}

void addEdge(int uu, int vv){
    v[ecnt] = vv;
    nxt[ecnt] = head[uu];
    head[uu] = ecnt;
    ecnt++;
}

void dfs(int x, int fa){
    if(d[x] > D) ret++;
    for(int e = head[x]; e != -1; e = nxt[e]) if(v[e] != fa){
        d[v[e]] = d[x] + 1;
        dfs(v[e], x);
    }
}

void solve(){
    d[0] = 0;
    dfs(0, -1);
    printf("%d\n", ret);
}

int main()
{
    int T, N, uu, vv;
    scanf("%d", &T);
    while(T--){
        init();
        scanf("%d%d", &N, &D);
        for(int i = 0; i < N-1; i++){
            scanf("%d%d", &uu, &vv);
            addEdge(uu, vv);
            addEdge(vv, uu);
        }
        solve();
    }
    return 0;
}