7-9 列出叶结点 (25 分)

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

输入格式：

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

输出格式：

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

输入样例：

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

输出样例：

4 1 5

 

源代码：

#include<iostream>
#include<queue>
using namespace std;
typedef struct TreeNode {
    int left, right;
}TreeNode, * Tree;
TreeNode p[100];
int change(char ch) {
    if (ch == '-')
        return -1;
    return ch - '0';
}
int main() {
    int n, i = 0, root, flag = 0;
    char a, b;
    queue<int> node;
    scanf("%d", &n);
    int* r = (int*)malloc(n * sizeof(int));
    for (int i = 0; i < n; i++)
        r[i] = 0;
    while (i < n) {
        cin >> a >> b;
        p[i].left = change(a);
        p[i].right = change(b);
        r[p[i].left] = 1;
        r[p[i].right] = 1;
        i++;
    }
    for (int i = 0; i < n; i++)
        if (!r[i])
            root = i;
    node.push(root);
    while (!node.empty()) {
        root = node.front();
        node.pop();
        if (p[root].left != -1)
            node.push(p[root].left);
        if (p[root].right != -1)
            node.push(p[root].right);
        if (flag == 0 && p[root].left == -1 && p[root].right == -1) {
            printf("%d", root);
            flag = 1;
        }
        else if (flag == 1 && p[root].left == -1 && p[root].right == -1)
            printf(" %d", root);
    }
    return 0;
}