PTA甲级习题和答案

树

AcWing 1476. 数叶子结点

可以用DFS,也可以用BFS
DFS(存结点编号,层数)

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

using namespace std;

const int N = 110;
int n, m, idx, max_depth; // idx 表示当前操作数的地址（非真正意义上的地址，是我们用于邻接表操作记录的地址; n是总节点数（包括根节点），m表示非叶子节点数，所以我们要读取m次数据，将其以链表的方式存储下来
int h[N], e[N], ne[N], cnt[N]; // h[x]表示每个单链表的头节点，e[x] 存储数值value， ne[x]存储下个元素的地址，cnt[x]存储这一深度有几个节点

// 在链表头插入
void add(int id, int v)
{
    e[idx] = v;
    ne[idx] = h[id];
    h[id] = idx ++;
}

void dfs(int id, int depth)
{
    if (h[id] == -1) // 如果走到底了，则头节点的下一个位置是-1
    {
        cnt[depth] ++;
        max_depth = max(depth, max_depth); // 更新最大深度用于后面的循环输出
        return;
    }
    for (int i = h[id]; ~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;//id和子结点的数量
        cin >> id >> k;
        // k表示以id为数值的非叶子节点下的子节点数量，所以我们要循环k次将子节点插入链表中
        while ( k -- )
        {
            int v;
            cin >> v;
            add (id, v); // 在头节点为id的链表中插入v
        }
    }
    dfs(1,0); // 从节点1开始深搜，初始高度为0
    cout << cnt[0];
    for (int i = 1; i <= max_depth; i ++ ) cout << " " << cnt[i] ;
    cout << endl;
    return 0;
}