PTA L2-011 玩转二叉树 (25 分)

题目链接：https://pintia.cn/problem-sets/994805046380707840/problems/994805065406070784

题意

        给定一棵二叉树的中序遍历和前序遍历，请你先将树做个镜面反转，再输出反转后的层序遍历的序列。所谓镜面反转，是指将所有非叶结点的左右孩子对换。这里假设键值都是互不相等的正整数。

思路

        根据中序遍历和前序遍历可以唯一创建一棵二叉树，先序遍历的第一个结点可将中序遍历的数据分成两个子序列，分别为这个结点的左子树的中序序列和右子树的中序序列，递归下去即可创建二叉树。
        镜面反转就是类似层序遍历二叉树，只不过是先遍历右孩子再遍历左孩子，与层序遍历相反。

代码

#include<bits/stdc++.h>
using namespace std;
const int maxn=35;

int n;
int pre[maxn],in[maxn];

struct node{
	int l,r;
}tree[maxn];

int Build(int la,int ra,int lb,int rb)
{
	//无孩子结点就输出0
	if(la>ra||lb>rb) return 0;
	int rt=pre[lb];
	int i=0;
	//找到中序序列里对应的结点
	while(in[la+i]!=rt) i++;
	tree[rt].r=Build(la+i+1,ra,lb+i+1,rb);
	tree[rt].l=Build(la,la+i-1,lb+1,lb+i);
	return rt;
}

void bfs(int rt)
{
	queue<int> q;
	q.push(rt);
	while(!q.empty()){
		int now=q.front();
		q.pop();
		if(tree[now].r>0) q.push(tree[now].r);
		if(tree[now].l>0) q.push(tree[now].l);
		printf("%d",now);
		if(!q.empty()) printf(" ");
	}
}

int main()
{
	scanf("%d",&n);
	for(int i=1;i<=n;i++) scanf("%d",&in[i]);
	for(int i=1;i<=n;i++) scanf("%d",&pre[i]);
	Build(1,n,1,n);
	bfs(pre[1]);
	return 0;
}