数据结构与算法实验5——树和二叉树 7-7 列出叶结点

7-7 列出叶结点 分数 25

对于给定的二叉树，本题要求你按从上到下、从左到右的顺序输出其所有叶节点。

输入格式：

首先第一行给出一个正整数 N（≤10），为树中结点总数。树中的结点从 0 到 N−1 编号。随后 N 行，每行给出一个对应结点左右孩子的编号。如果某个孩子不存在，则在对应位置给出 "-"。编号间以 1 个空格分隔。

输出格式：

在一行中按规定顺序输出叶节点的编号。编号间以 1 个空格分隔，行首尾不得有多余空格。

输入样例：

8
1 -
- -
0 -
2 7
- -
- -
5 -
4 6

输出样例：

4 1 5

代码长度限制

16 KB

时间限制

400 ms

内存限制

64 MB

#include <bits/stdc++.h>
using namespace std;
const int N = 1e5 + 10;

struct node
{
    int l, r;
} tree[N];

int vis[N]; //值为一表示不是根节点

queue<int> q;
vector<int> leaf;

void bfs() // 层序遍历寻找叶子节点
{
    while (q.size())
    {
        int t = q.front();
        q.pop();
        if (tree[t].l == -1 && tree[t].r == -1)
            leaf.push_back(t); //存放叶子节点
        if (tree[t].l != -1)
            q.push(tree[t].l);
        if (tree[t].r != -1)
            q.push(tree[t].r);
    }
}

int main()
{
    int n;
    cin >> n;
    char a, b;
    for (int i = 0; i < n; i++)
    { // 建树
        cin >> a >> b;
        if (a == '-')
            tree[i].l = -1;
        else
        {
            tree[i].l = a - '0'; // i是a的父亲节点
            vis[a - '0'] = 1;    //标记a不是根节点，因为根节点没有父亲节点
        }
        if (b == '-')
            tree[i].r = -1;
        else
        {
            tree[i].r = b - '0'; // i是b的父亲节点
            vis[b - '0'] = 1;    //标记b不是根节点
        }
    }
    int root;
    for (int i = 0; i < n; i++) //寻找根节点
    {
        if (!vis[i])
        {
            root = i; //根节点
            break;
        }
    }
    q.push(root);
    bfs();
    bool flag = 0;
    for (auto i : leaf) //输出所有的叶子节点
    {
        if (flag)
            cout << " ";
        else
            flag = 1;
        cout << i;
    }
    return 0;
}