PTA列出叶结点(超详解)

7-38 列出叶结点 (300 分)
对于给定的二叉树,本题要求你按从上到下、从左到右的顺序输出其所有叶节点。

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

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

输入样例:
8
1 -

0 -
2 7

5 -
4 6
输出样例:
4 1 5
存储:本题所给数据方式便于存储父亲编号,但是左右节点都指向父亲节点,便于区分左右,左孩子存储父节点编号的相反数。右孩子直接存储父亲编号,但是0的左右孩子存储±0无区别,故编号1-N合适。
遍历:题目要求从上至下,从左至右,使用quene队列层序遍历,如上存储结果,根节点存储0,找到根压入队列,层序遍历,如果找到没有子节点的节点(没有节点的绝对值指向自己编号)就输出改编号减一即可

数据演示如下
PTA列出叶结点(超详解)_第1张图片

//让编号从1到N,pre存储父节点,如pre[5] = -8 pre[7] = 8表示 8 8->L=5 8->R=7;
#include 
using namespace std;
int main() {
	int n;
	char nA, nB;
	vector<int> pre(10000, 0);
	queue<int> q;
	bool falgA = false;
	cin >> n;
	for (int i = 1; i <= n; i++) {
		cin >> nA >> nB;
		if(nA!='-'){
			pre[(nA-'0') + 1] = -i;//左孩子标记父亲
		}
		if (nB != '-') {
			pre[(nB - '0') + 1] = i;//右孩子标记父亲
		}
	}
	for (int i = 1; i <= n; i++) {//找根
		if (!pre[i]) {
			q.push(i);
			pre[i] = 0;
			break;
		}
	}
	while (!q.empty()){
		bool falgB = true;//标记该节点为叶节点
		for (int i = 1; i <= n; i++) {//找左孩子
			if (pre[i] == -q.front()) {
				q.push(i);
				pre[i] = 0;
				falgB = false;
			}
		}
		for (int i = 1; i <= n; i++) {//找右孩子
			if (pre[i] == q.front()) {
				q.push(i);
				pre[i] = 0;
				falgB = false;
			}
		}
		if (falgB) {//该节点是叶节点
			if (!falgA) {
				cout << q.front() - 1;
				falgA = true;
			}
			else {
				cout << " " << q.front() - 1;
			}
		}
		q.pop();
	}
}

你可能感兴趣的:(算法)