PAT 1004 Counting Leaves（BFS+模拟树）

原题地址

https://www.patest.cn/contests/pat-a-practise/1004

题意：一棵有N个结点的树，其中有M个结点是非叶子结点，给出每个非叶子结点的孩子结点编号，求出每一层上的非叶子结点个数。

解题思路

本题考察树的结构，这种看似树结构，但用数字编号表示节点的题目，经常可以用数组模拟，无需构造出一棵树。

数据结构：

　　注意到结点编号最多为两位，因此100以内的数组就足以代表这棵树，用结构体数组a[i]表示编号为i的节点的信息（a[1]固定为根节点），结构体的成员cnt代表有多少个孩子，成员child[]记录它孩子的编号。那么用下标就可以索引到这个节点啦。

算法：

　　很容易就想到用BFS来做每一层的计数。
　　初始时队列里存放根节点，每次while循环里完成两件事情，第一件就是统计这一层有多少叶子结点，第二件就是把每个节点的孩子放入队尾待下一次while循环处理。实际操作时注意此次访问队列前确定队列的大小（即这一层的节点总数），防止push以后改变了队列的大小而遍历出界。

AC代码

#include <iostream>
#include <queue>
using namespace std;
const int maxn = 105; //题目保证结点编号是两位数

typedef struct node
{
    int cnt; //孩子节点的个数
    int child[maxn];
}NODE;

NODE a[maxn];

void BFS()
{
    queue<int> q; //队列，用于存放这一层的结点
    q.push(1);
    int ans[maxn] = {0}, lev = 0, tmp;
    while (!q.empty())
    {
        int sz = q.size(), thisLevel = 0; //记录这层的叶子结点总数
        for (int i = 0; i<sz; ++i)
        {
            tmp = q.front(); q.pop();
            if (a[tmp].cnt == 0)
                thisLevel++;   //tmp是这一层的叶子结点
            for (int j = 0; j < a[tmp].cnt; ++j)
                q.push(a[tmp].child[j]); //tmp的所有叶子结点入队
        }
        ans[lev++] = thisLevel;
    }
    for (int i = 0; i < lev; ++i)
    {
        if (i) cout << ' ';
        cout << ans[i];
    }
}

int main()
{
    ios::sync_with_stdio(false);
    int n, m;
    cin >> n >> m;
    for (int i = 0; i < maxn; ++i) //结点数组初始化
        a[i].cnt = 0;
    //输入所有非叶子节点的孩子信息
    int id, k, tmp;
    for (int i = 0; i<m; ++i)
    {
        cin >> id >> k;
        a[id].cnt = k;
        for (int j = 0; j < k; ++j)
            cin >> a[id].child[j];
    }
    BFS();
    return 0;
}

算法复杂度：O(N)
耗时：0 ms