PAT 数叶子节点（1004）

最新推荐文章于 2025-05-26 09:02:48 发布

Simod chen

最新推荐文章于 2025-05-26 09:02:48 发布

阅读量167

点赞数

分类专栏：考研算法PAT 文章标签：链表数据结构算法二叉树 dfs

本文链接：https://blog.youkuaiyun.com/Chenxiaoyu99/article/details/115028187

版权

考研算法PAT 专栏收录该内容

4 篇文章

订阅专栏

本文介绍了一种使用C++实现的算法，通过邻接表构造树，并递归地遍历树来统计每层的叶子节点数量。文章展示了完整的代码示例，解释了如何通过输入节点及其子节点来构建树结构，最后输出每层叶子节点的数量。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

PAT 数叶子节点（1004）

#include <iostream>
#include <cstring>
#include <algorithm>


using namespace std;

const int N = 110;

int n, m;
int h[N], e[N], ne[N], idx;//该地方利用了数组模拟邻接表的原理来构造树，可以多查找相关的表示方法，网上有很多。
int cnt[N], max_depth;
void add(int a, int b)
{
    e[idx] = b, ne[idx] = h[a], h[a] = idx ++;//e数组存的是该节点内容
    //ne数组表示与a相连的下一个节点而这一切都是由idx所相联系在一起的
    
}
void dfs(int u, int depth)
{
    if(h[u] == -1)//表示该节点没有孩子节点因为以u为链表指向-1
    {
        cnt[depth] ++ ;//cnt数组表示某一层有多少个叶子节点的计数器
        max_depth = max(max_depth, depth);//计算最大的树深度
        return;
    }
    
    for(int i = h[u]; ~i; i = ne[i])
        dfs(e[i], depth + 1);
    
    
}


int main()
{
    cin >> n >> m;
    
    memset(h, -1, sizeof h);
    
    for(int i = 0; i < m; i ++ )
    {
        int id, k;
        cin >> id >> k;
        while(k -- )
        {
            int son;
            cin >> son;
            add(id, son);
        }
    }
    
    dfs(1, 0);
    
    cout << cnt[0];
    for(int i = 1; i <= max_depth; i ++ ) cout << ' ' << cnt[i];//PAT容易卡结果的空格所以要规范输出
    
    cout << endl;
    
    return 0;
    
    
    
    
}```