PAT甲级真题1004 数叶子结点(DFS,BFS两种解法)

家庭关系可以用家谱树来表示，给定一个家谱树，你的任务是找出其中没有孩子的成员。

输入格式
第一行包含一个整数 NN 表示树中结点总数以及一个整数 MM 表示非叶子结点数。

接下来 MM 行，每行的格式为：

ID K ID[1] ID[2] … ID[K]
IDID 是一个两位数字，表示一个非叶子结点编号，KK 是一个整数，表示它的子结点数，接下来的 KK 个 ID[i]ID[i] 也是两位数字，表示一个子结点的编号。

为了简单起见，我们将根结点固定设为 0101。

所有结点的编号即为 01,02,03,…,31,32,33,…,N01,02,03,…,31,32,33,…,N。

输出格式
输出从根结点开始，自上到下，树的每一层级分别包含多少个叶子节点。

输出占一行，整数之间用空格隔开。

数据范围
0<N<1000<N<100
输入样例：
2 1
01 1 02
输出样例：
0 1
样例解释
该样例表示一棵只有 22 个结点的树，其中 0101 结点是根，而 0202 结点是其唯一的子节点。

因此，在根这一层级上，存在 00 个叶结点；在下一个级别上，有 11 个叶结点。

所以，我们应该在一行中输出0 1。

//DFS
#include <cstring>
#include <iostream>
#include <algorithm>

using namespace std;

int n, m;
vector<int>v[110];
int cnt[110], max_depth;//记录每层叶子结点个数和最大深度
void dfs(int u, int depth)
{
    if (v[u].size() == 0)  // 说明u是叶子节点
    {
        cnt[depth] ++ ;
        max_depth = max(max_depth, depth);
        return;
    }

    for (int i = 0; i<v[u].size(); i++){
        int e=v[u][i];
        dfs(e, depth + 1);
    }
}

int main()
{
    cin >> n >> m;
    for (int i = 0; i < m; i ++ )
    {
        int id, k;
        cin >> id >> k;
        while (k -- )
        {
            int son;
            cin >> son;
            v[id].push_back(son);
        }
    }

    dfs(1, 0);
    cout << cnt[0];
    for (int i = 1; i <= max_depth; i ++ ) cout << ' ' << cnt[i];
    cout << endl;

    return 0;
}

//BFS
#include <cstring>
#include <iostream>
#include <algorithm>
#include <vector>
#include <queue>
using namespace std;

int n, m;
vector<int>v[110];
int cnt[110], max_depth;//记录每层叶子结点个数和最大深度
int high[110];//记录每个节点的深度
void bfs(int u)
{
    queue<int>q;
    q.push(u);
    high[u]=0;
    while(q.size()){
        int top=q.front();
        q.pop();
        max_depth = max(max_depth,high[top]);
        if (v[top].size() == 0)  // 说明u是叶子节点
        {
        cnt[high[top]] ++ ;
        continue;
      }
    for (int i = 0; i<v[top].size(); i++){
        int e=v[top][i];
        high[e]=high[top]+1;
        q.push(e);
      }
   }
}


int main()
{
    cin >> n >> m;
    for (int i = 0; i < m; i ++ )
    {
        int id, k;
        cin >> id >> k;
        while (k -- )
        {
            int son;
            cin >> son;
            v[id].push_back(son);
        }
    }

    bfs(1);
    cout << cnt[0];
    for (int i = 1; i <= max_depth; i ++ ) cout << ' ' << cnt[i];
    cout << endl;

    return 0;
}