对于给定的二叉树,本题要求你按从上到下、从左到右的顺序输出其所有叶节点。
输入格式:
首先第一行给出一个正整数 N(≤10),为树中结点总数。树中的结点从 0 到 N−1 编号。随后 N 行,每行给出一个对应结点左右孩子的编号。如果某个孩子不存在,则在对应位置给出 “-“。编号间以 1 个空格分隔。
输出格式:
在一行中按规定顺序输出叶节点的编号。编号间以 1 个空格分隔,行首尾不得有多余空格。
输入样例:
8
1 -
- -
0 -
2 7
- -
- -
5 -
4 6
输出样例:
4 1 5
#include<map>
#include<cstdio>
#include<iostream>
#include<string>
#include<algorithm>
#include<cstdlib>
#include<queue>
using namespace std;
const int maxn = 1e3 + 10;
int lchi[maxn], rchi[maxn], inCnt[maxn] = { 0 }, outCnt[maxn] = { 0 }, flag = 0;
void init(int n) {
for (int i = 0; i <= n; i++) lchi[i] = rchi[i] = -1;
}
void bfs(int root) {
queue<int>Q;
Q.push(root);
while (!Q.empty()) {
int u = Q.front(); Q.pop();
if (outCnt[u] == 0) {
if (flag) printf(" ");
else flag = 1;
printf("%d", u);
}
if (lchi[u] != -1) Q.push(lchi[u]);
if (rchi[u] != -1) Q.push(rchi[u]);
}
}
int main() {
int n,root; cin >> n;
init(n);
for (int i = 0; i < n; i++) {
char u, v; cin >> u >> v;
if (u != '-') lchi[i] = u - '0', outCnt[i]++, inCnt[u - '0']++;
if (v != '-') rchi[i] = v - '0', outCnt[i]++, inCnt[v - '0']++;
}
for (int i = 0; i < n; i++) if (inCnt[i] == 0) root = i;
bfs(root);
}