HDU5326多校签到题，搜索子孙节点，

最新推荐文章于 2024-07-22 22:20:26 发布

翻译最新推荐文章于 2024-07-22 22:20:26 发布 · 436 阅读

数据结构专题同时被 2 个专栏收录

63 篇文章

订阅专栏

HDU

56 篇文章

订阅专栏

本文介绍了一种使用深度优先搜索（DFS）遍历树形结构的方法，并通过具体代码实现展示了如何统计特定子树节点数量的过程。该算法适用于解决与树结构相关的计数问题。

dfs一发即可：

#include <bits/stdc++.h>
using namespace std;
const int N = 110;
vector<int> E[N];
int in[N], son[N];
bool vis[N];
void dfs(int u)
{
    vis[u] = true;
    son[u] = 1;
    for(int v: E[u]) {
        dfs(v);
        son[u] += son[v];
    }
}
int main()
{
    freopen("in.txt", "r", stdin);
    freopen("out.txt", "w", stdout);
    int n, k, a, b;
    while(scanf("%d%d", &n, &k) == 2) {
        memset(vis, false, sizeof(vis));
        memset(in, 0, sizeof(in));
        for(int i = 0; i < n; i++) {
            E[i].clear();
        }
        for(int i = 1; i < n; i++) {
            scanf("%d%d", &a, &b);
            assert(1 <= a && a <= n);
            assert(1 <= b && b <= n);
            a--; b--;
            E[a].push_back(b);
            in[b]++;
        }
        int root = -1;
        for(int i = 0; i < n; i++) if(!in[i]) {
            root = i;
        }
        assert(root == 0);
        dfs(root);
        for(int i = 0 ; i < n; i++) if(!vis[i]) {
            while(1);
        }
        int ret = 0;
        for(int i = 0; i < n; i++) if(son[i] - 1 == k) {
            ret ++;
        }
        printf("%d\n", ret);
    }
    return 0;
}