PTA 列出叶结点

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

输入格式：

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

输出格式：

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

输入样例：

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

输出样例：

4 1 5

思路：

使用一个结构体来存储一个点的左右节点；

每个点对应的左右节点，用char型进行输入，如果是'-'就当作-1，否则就当成ch-'0';

使用队列进行遍历，如果左右节点都是-1，就存入vector中，最后输出；

代码：

#include<bits/stdc++.h>
using namespace std;
int n;
struct node{
	int r , l ;
}s[10];
vector<int>ans;
void run(int x){
	queue<int>q;
	q.push(x);
	while(!q.empty()){
		int t = q.front();
		if(t == -1){
			q.pop();
			continue;
		}
		if(s[t].l == -1 && s[t].r == -1)ans.push_back(t);
		q.pop();
		 q.push(s[t].l);
		 q.push(s[t].r);
	}
}
int main(){
   cin >> n;
   map<int,int>mp;
   for(int i = 0 ; i < n ; i ++){
		char a , b;
		cin >> a >> b;
		if(a == '-')s[i].l = -1;
		else s[i].l = a - '0';
		if(b == '-')s[i].r = -1;
		else s[i].r = b - '0';
		mp[a-'0'] = mp[b-'0'] = 1;
   }
   for(int i = 0 ; i < n ; i ++){
		if(!mp[i]){
			run(i);
			break;
		}
   }
   for(int i = 0 ; i < ans.size() ; i ++){
   		cout << ans[i];
   		if(i != ans.size() - 1)cout << " ";
   		else cout << endl;
   }
}