【题解】【AcWing】1497. 树的遍历

1497. 树的遍历

原题传送：AcWing 1497. 树的遍历

一个二叉树，树中每个节点的权值互不相同。

现在给出它的后序遍历和中序遍历，请你输出它的层序遍历。

输入格式

第一行包含整数 $N$ ，表示二叉树的节点数。

第二行包含 $N$ 个整数，表示二叉树的后序遍历。

第三行包含 $N$ 个整数，表示二叉树的中序遍历。

输出格式

输出一行 $N$ 个整数，表示二叉树的层序遍历。

数据范围

$1\le N\le 30$

输入样例：

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

输出样例：

4 1 6 3 5 7 2

思路：

后序遍历和中序遍历构造树，然后bfs进行层序遍历。

题解：

#include<bits/stdc++.h>

using namespace std;

const int N = 40;

int n;
int postorder[N], inorder[N];
unordered_map<int, int> l, r, pos;
int q[N];

int build(int il, int ir, int pl, int pr)
{
	int root = postorder[pr];
	int k = pos[root];
	if(il < k)
		l[root] = build(il, k - 1, pl, pl + (k - 1 - il));
	if(k < ir)
		r[root] = build(k + 1, ir, pl + (k - 1 - il) + 1, pr - 1);
	return root;
}

void bfs(int root)
{
	int hh =  0, tt = 0;
	q[0] = root;
	
	while(hh <= tt)
	{
		int t = q[hh++];
		if(l.count(t))
			q[++tt] = l[t];
		if(r.count(t))
			q[++tt] = r[t];
	}
	
	cout << q[0];
	for(int i = 1; i < n; i++)
		cout << " " << q[i];
	cout << endl;
}

int main()
{	
	cin >> n;
	
	for(int i = 0; i < n; i++)
		cin >> postorder[i];
	for(int i = 0; i < n; i++)
	{
		cin >> inorder[i];
		pos[inorder[i]] = i;
	}
		
	int root = build(0, n - 1, 0, n - 1);
	
	bfs(root);
	
	return 0;
}