对于给定的二叉树,本题要求你按从上到下、从左到右的顺序输出其所有叶节点。
输入格式:
首先第一行给出一个正整数 N(≤10),为树中结点总数。树中的结点从 0 到 N−1 编号。随后 N 行,每行给出一个对应结点左右孩子的编号。如果某个孩子不存在,则在对应位置给出 “-”。编号间以 1 个空格分隔。
输出格式:
在一行中按规定顺序输出叶节点的编号。编号间以 1 个空格分隔,行首尾不得有多余空格。
输入样例:
8
1 -
- -
0 -
2 7
- -
- -
5 -
4 6
输出样例:
4 1 5
#include <iostream>
#include <map>
#include <queue>
using namespace std;
struct Node {
int data, L, R;
};
int N, root, indegree[15], flag = 0;
map<int, Node>mp;
int main() {
cin >> N;
string a, b;
vector<char>v(N);
for (int i = 0; i < N; ++i) {
cin >> a >> b;
mp[i] = { i, a == "-" ? 10 : stoi(a), b == "-" ? 10 : stoi(b) };
indegree[mp[i].L]++, indegree[mp[i].R]++;
}
for (int i = 0; i < N; ++i)
if (!indegree[i]) {
root = i;
break;
}
queue<Node>q;
q.push(mp[root]);
while (q.size()) {
Node tmp = q.front();
q.pop();
if (tmp.L == 10 && tmp.R == 10)
cout << (flag++ ? " " : "") << tmp.data;
if (tmp.L < 10)
q.push(mp[tmp.L]);
if (tmp.R < 10)
q.push(mp[tmp.R]);
}
return 0;
}
版权声明:本文为qq_45798109原创文章,遵循CC 4.0 BY-SA版权协议,转载请附上原文出处链接和本声明。