L2-006. 树的遍历 （递归建树）

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 <iostream>
#include <algorithm>
#include <vector>
using namespace std;
vector<int> in; 中序数组
vector<int> post; 后序数组
vector<int> tree(100000);

void built(int pl,int pr, int il, int ir,int k) // pr -> post_right  ; ir -> in_right ; K为当前元素在数组中的下标
{
	if (il > ir || pl > pr)
		return;
	int father = post[pr];
	int index = distance(in.begin(), find(in.begin(), in.end(), father));
	int len = index - il;
	built(pl, pl + len - 1, il, index - 1,2 * k);
	built(pl + len, pr - 1, index + 1, ir, 2 * k + 1);
	tree[k] = father;
}



int main()
{
	//freopen("data.txt", "r", stdin);
	int N;
	int num;
	int flag = true;
	cin >> N;
	for (int i = 0; i < N; i++) {
		cin >> num;
		post.push_back(num);
	}
	for (int i = 0; i < N; i++) {
		cin >> num;
		in.push_back(num);
	}
	built(0, N - 1, 0, N - 1, 1);
	int cnt = 0;
	for (int i = 1;; i++) 
	{
		if (cnt >= N)
			return 0;
		if (tree[i] == 0)
			continue;
		else
		{
			if (flag) {
				cout << tree[i];
				flag = false;
			}
			else
				cout << " " << tree[i];
			cnt++;
		}
	}
	return 0;
}