L2-011. 玩转二叉树

L2-011. 玩转二叉树

时间限制

400 ms

内存限制

65536 kB

代码长度限制

8000 B

判题程序

Standard

作者

陈越

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

输入格式：

输入第一行给出一个正整数N（<=30），是二叉树中结点的个数。第二行给出其中序遍历序列。第三行给出其前序遍历序列。数字间以空格分隔。

输出格式：

在一行中输出该树反转后的层序遍历的序列。数字间以1个空格分隔，行首尾不得有多余空格。

输入样例：

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

输出样例：

4 6 1 7 5 3 2

---

提交代码

分析：给出中序和前序，通过前序确定根结点，然后在中序中找左子树和右子树，递归构建出树。输出的时候因为是镜面，所以遍历的时候先右后左

AC代码：

#include<bits/stdc++.h>
using namespace std;
struct node
{
	int k,l,r;
}p[1000];
int in_order[1000];
int pre_order[1000];
int buildtree(int l,int r,int ll,int rr)
{
	int L;
	if(l>r) return -1;
	p[ll].k=pre_order[ll];
	for(int i=l;i<=r;i++)
	{
		if(in_order[i]==p[ll].k)
		{
			L=i;
			break;
		}
	}
	p[ll].l=buildtree(l,L-1,ll+1,ll+L-l);
	p[ll].r=buildtree(L+1,r,ll+L-l+1,rr);
	//printf("%d %d %d\n",ll,p[ll].l,p[ll].r);
	return ll;
}
void bfs()
{
	queue<int>q;
	q.push(0);
	int flag=0;
	while(!q.empty())
	{
		int pp=q.front();
		q.pop();
		if(!flag)
		printf("%d",p[pp].k);
		else
		printf(" %d",p[pp].k);
		flag=1;
		if(p[pp].r!=-1)
		q.push(p[pp].r);
		if(p[pp].l!=-1)
		q.push(p[pp].l);
	}
	printf("\n");
}
int main()
{
	int n;
	scanf("%d",&n);
	for(int i=0;i<n;i++)
	{
		scanf("%d",&in_order[i]);
	}
	for(int i=0;i<n;i++)
	{
		scanf("%d",&pre_order[i]);
	}
	buildtree(0,n-1,0,n-1);
	bfs();
}