1004 Counting Leaves (30 分) bfs

最新推荐文章于 2026-06-23 22:46:37 发布

原创最新推荐文章于 2026-06-23 22:46:37 发布 · 86 阅读

0 ·

本内容遵循CC 4.0 BY-SA版权协议

标签

#算法 #c++ #数据结构 #bfs #stl

该博客主要介绍了如何利用C++实现广度优先搜索（BFS）来遍历非二叉树结构。代码中通过vector容器存储树结构，并用层次遍历的方式计算每个层级的节点数。程序首先读取树的节点和边信息，然后从根节点开始进行BFS。在遍历过程中，更新最大层级和各层级节点数量。最后输出结果。该算法适用于处理非二叉树的层次信息。

这个题吧用vector容器存储树
这个树他不是二叉树所以直接用vector定义二维数组
然后每到一层就push_back()把他加进去
ans是保存最终答案的
leavel是保存层数的
maxleavel是保存一共有多少层的
` if(tree[t].size()==0){

ans[leavel[t]]++;
    }`
    核心代码，如果他的没有孩子就让ans对应的层数++

剩下的就是广搜bfs的标准模板了

#include<iostream>
#include<vector>
#include<algorithm>
#include<queue>
using namespace std;
vector<int>tree[1010];
int ans[1010];
int leavel[1010];
int maxlevel=-1;
queue<int>T;
void bfs(int begin) {
    T.push(begin);
    int t;
    leavel[begin]=0;
    while (!T.empty()) {
        t = T.front();
        T.pop();
        maxlevel=max(leavel[t],maxlevel);
        if(tree[t].size()==0){
            ans[leavel[t]]++;
        }
        for (int i = 0; i <tree[t].size(); i++) {
            leavel[tree[t][i]]=leavel[t]+1;
            T.push(tree[t][i]);
        }
    }
}
int main()
{
    int n, m;
    cin >> n >> m;
    for (int i = 0; i < m; i++) {
        int a, k;
        cin >> a >> k;
        for (int j = 0; j < k; j++) {
            int b;
            cin >> b;
            tree[a].push_back(b);
        }
    }
    bfs(1);
    for (int i = 0; i <=maxlevel; i++) {
        if (i != 0) {
            cout << " ";
        }
        cout << ans[i];
    }
    return 0;
}