(L2-011)玩转二叉树(利用个给定序列建树)

题目链接：PTA | 程序设计类实验辅助教学平台

分析：这道题分为三个步骤：

（1）根据中序遍历序列和前序遍历序列进行建树

（2）将树进行反转

（3）对树进行层序遍历

先来说一下如何根据中序遍历序列和前序遍历序列进行建树，由前序遍历的方式我们可以知道，先遍历根然后依次遍历左孩子和右孩子，由于建树我们肯定是先建立根再依次建立孩子节点，所以我们直接按照给定的前序遍历进行一个一个插入树中即可（一开始为空树），而根节点刚好把中序遍历分为三部分，根左边的为左子树，根，根右边的为右子树，这样我们就可以进行递归建树了，当某一部分长度为1时就可以直接进行赋值。

下面来说一下反转，这个比较简单，就是直接按照后序遍历的方式，然后直接将左右孩子交换即可

最后是层序遍历，这个是用队列实现的，详情见代码：

#include<cstdio>
#include<iostream>
#include<cstring>
#include<vector>
#include<algorithm>
#include<map>
#include<cmath>
#include<queue>
using namespace std;
const int N=1e5+10;
int a[N],b[N],t[N];
int l[N],r[N],idx;
bool vis[N];
void build(int pos,int L,int R,int val)
{
	if(!vis[pos]||L==R)
	{
		t[pos]=val;
		vis[pos]=true;
		return ;
	}
	int i,j;
	for(i=L;i<=R;i++)//找待插入值在中序遍历中的位置 
		if(a[i]==val) break;
	for(j=L;j<=R;j++)//找当前值在中序遍历中的位置
		if(a[j]==t[pos]) break;
	if(i<j)//说明待插入节点在以当前节点为根的左子树中 
	{
		if(l[pos]) build(l[pos],L,j-1,val);
		else l[pos]=++idx,build(l[pos],L,j-1,val);
	}
	else//说明待插入节点在以当前节点为根的右子树中
	{
		if(r[pos]) build(r[pos],j+1,R,val);
		else r[pos]=++idx,build(r[pos],j+1,R,val);
	}
}
void dfs(int pos)//反转 
{
	if(l[pos]) dfs(l[pos]);
	if(r[pos]) dfs(r[pos]);
	swap(l[pos],r[pos]);
}
void bfs(int x)//求层序遍历 
{
	queue<int>q;
	q.push(x);
	printf("%d",t[x]);
	while(!q.empty())
	{
		int tt=q.front();
		if(tt!=x)
			printf(" %d",t[tt]);
		q.pop();
		if(l[tt]) q.push(l[tt]);
		if(r[tt]) q.push(r[tt]);
	}
}
int main()
{
	int n;
	cin>>n;
	for(int i=1;i<=n;i++)
		scanf("%d",&a[i]);
	idx=1;
	for(int i=1;i<=n;i++)
	{
		scanf("%d",&b[i]);
		build(1,1,n,b[i]);
	}
	dfs(1);
	bfs(1);
	return 0;
}