团体程序设计天梯赛-练习集 L2-006 树的遍历 根据后序中序建树+BFS

L2-006. 树的遍历

时间限制

400 ms

内存限制

65536 kB

代码长度限制

8000 B

判题程序

Standard

作者

陈越

给定一棵二叉树的后序遍历和中序遍历，请你输出其层序遍历的序列。这里假设键值都是互不相等的正整数。

输入格式：

输入第一行给出一个正整数N（<=30），是二叉树中结点的个数。第二行给出其后序遍历序列。第三行给出其中序遍历序列。数字间以空格分隔。

输出格式：

在一行中输出该树的层序遍历的序列。数字间以1个空格分隔，行首尾不得有多余空格。

输入样例：

7
2 3 1 5 7 6 4
1 2 3 4 5 6 7

输出样例：

4 1 6 3 5 7 2

#include <cstdio>
#include <cstring>
#include <iostream>
#include <map>
#include <algorithm>
#include <queue>

using namespace std;

const int maxn = 30;
int a[maxn + 10], b[maxn + 10];
map<int, int>L, R; //L[i]表示i的左子节点,R[i]表示i的右子节点

//当前考虑的是后序遍历[la, ra]区间和对应的中序遍历[lb,rb]区间
int build(int la, int ra, int lb, int rb) {
	if (la > ra) { //越界判断
		return 0;
	}
	int root = a[ra]; //后序遍历右一肯定是根
	int i;
	for (i = lb; i <= rb && b[i] != root; i++) {} //在中序遍历中找到这个节点下标i
	if (i <= rb) { //递归分治
		R[root] = build(ra - rb + i, ra - 1, i + 1, rb);
		L[root] = build(la, ra - rb + i - 1, lb, i - 1);
	}
	return root;
}

void bfs(int root) {
	queue<int> Q;
	Q.push(root);

	int cnt = 0;
	while (!Q.empty()) {
		int tn = Q.front(); Q.pop();
		printf(cnt++ == 0 ? "%d" : " %d", tn);
		if (L[tn]) {
			Q.push(L[tn]);
		}
		if (R[tn]) {
			Q.push(R[tn]);
		}
	}
	puts("");
}

int main()
{
	int N;
	scanf("%d", &N);
	for (int i = 1; i <= N; i++) {
		scanf("%d", &a[i]);
	}
	for (int i = 1; i <= N; i++) {
		scanf("%d", &b[i]);
	}
	int root = build(1, N, 1, N);
	bfs(root);
	return 0;
}