7-4 树的遍历 (20分)

7-4 树的遍历 (20分)

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

输入格式：

输入第一行给出一个正整数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

思路：

首先不就是通过中序后序遍历来还原二叉树嘛，和上一周的还原二叉树一个思路，不同的就是层序遍历，这里我用了书上的用队列实现二叉树的层序遍历的方法

遍历从二叉树的根结点开始，首先将根结点指针入队列，然后从队头取出一个元素，每取一个元素，执行以下操作：(1)访问该元素所指结点 (2)若该元素所指结点的左、右孩子结点非空，则将该元素所指结点的左孩子指针和右孩子指针顺序入队

AC代码：

#include<bits/stdc++.h>
using namespace std;
typedef struct btnode{
	int data;
	struct btnode *left,*right;
}BTnode,*BTree;
int inod[100],postod[100];
BTnode* Retree(int root,int begin,int end){		//root是后序中的根结点，在后序的末尾，begin是中序的开头，end是中序的结尾 
	if(begin > end)
		return NULL;
	BTnode *node = new BTnode;
	node->data = postod[root];
	int i;
	for(i = begin;inod[i]!=postod[root];i++);	//找到中序中根结点的位置 
	node->left = Retree(root - 1 - (end - i),begin,i-1);
	node->right = Retree(root - 1,i+1,end);
	return node;
}
void bfs(BTree bt) {	//通过队列实现二叉树的层序遍历
	BTnode *queue[100];
	int front = -1,rear = 0;
	int flag = 0;
	if(bt == NULL)
		return;
	queue[rear] = bt;
	while(front != rear){	//当front和rear相等时队列为空
		front++;
		if(flag++)
		cout << " ";
		cout << queue[front]->data;
		if(queue[front]->left != NULL){
			rear++;
			queue[rear] = queue[front]->left;
		}
		if(queue[front]->right != NULL){
			rear++;
			queue[rear] = queue[front]->right;
		}
	}
}
int main(){
	int n;
	cin >> n;
	for(int i = 0;i < n;i++)
		cin >> postod[i];
	for(int i = 0;i < n;i++)
		cin >> inod[i];
	bfs(Retree(n-1,0,n-1));//注意后序遍历的root应该传入最后一个值才是根结点，与先序恰反 
}

大佬的精简代码：

#include<iostream>
#include<vector>
using namespace std;
vector<int>post, in, level(100000, -1);
int N;
void ergodic(int root, int start, int end, int index) {
	int i = start;
	if (start > end)
		return;
	while (i < end&&in[i] != post[root])
		i++;
	level[index] = post[root];
	//cout << root << start << end << i << endl;
	ergodic(root - 1 - end + i, start, i - 1, 2 * index + 1);
	ergodic(root - 1, i + 1, end, 2 * index + 2);
}
int main() {
	cin >> N;
	post.resize(N);
	in.resize(N);
	for (int i = 0; i < N; i++)
		cin >> post[i];
	for (int i = 0; i < N; i++)
		cin >> in[i];
	ergodic(N - 1, 0, N - 1, 0);
	for (int i = 0, cnt = 0; i < level.size(); i++) {
		if (level[i] != -1) {
			cout << level[i];
			cnt != N - 1 ? cout << ' ' : cout << endl;
			cnt++;
		}
	}
	return 0;
}