PAT - 天梯赛 L2-006 树的遍历

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

输入格式：

输入第一行给出一个正整数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<queue>
using namespace std;
int hou[31];
int mid[31];
int n;
struct node{
	int l;
	int r;
}tree[31];
int build(int la,int ra,int lb,int rb){//重建tree la,ra代表中序，lb,rb代表后序； 
	if(la>ra)return -1;
	int root=hou[rb],p1,p2;
	p1=la;
	while(mid[p1]!=root)p1++;
	p2=p1-la;//中间节点的左子代个数 
	tree[root].l=build(la,p1-1,lb,lb+p2-1);
	tree[root].r=build(p1+1,ra,lb+p2,rb-1);
	return root;
}
void bfs(int a){
	queue<int>p;
	p.push(a);
	int co=0;
	while(!p.empty()){
		int f=p.front();
		p.pop();
		if(co != n-1){//最后一个不输出空格 
			printf("%d ",f);
			co++;
		}else{
			printf("%d",f);
			return;
		}
		if(tree[f].l!=-1){
			p.push(tree[f].l);
		}
		if(tree[f].r!=-1){
			p.push(tree[f].r);
		}
	}
}
int main(){
	scanf("%d",&n);
	for(int i = 0; i < n; i++){
		scanf("%d", &hou[i]);
	}
	for(int i = 0; i < n; i++){
		scanf("%d", &mid[i]);
	}
	int root=build(0,n-1,0,n-1);
	bfs(root);
	return 0;
}