天梯赛 L2-011. (二叉树) 玩转二叉树

时间:2023-03-09 05:57:55
天梯赛   L2-011.   (二叉树)    玩转二叉树

题目链接

题目描述

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

输入格式:

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

分析:

给定一棵二叉树的前序和中序序列,可以唯一的确定一棵二叉树。

所谓的层次遍历的序列就是要按照从上到下从左到右的顺序输出来。但这里眼球镜面反转后在输出层次遍历的序列,我们一般在根据前序和中序确定一棵树的时候,先递归这棵树的左子树再递归右子树,但如果我们现在先递归右子树再递归左子树,这样每次递归的根节点的出现序列就是镜面反转后的序列。

代码:

#include <iostream>
#include<cstring>
#include<stdio.h>
using namespace std;
int post[35],in[35],pre[35],level[35];
int N=0;
//l1~r1前序的区间,l2~r2中序的区间
void postTranverse(int l1,int r1,int l2,int r2,int id)
{
if(l1>r1||l2>r2) return;
int i,len1,len2;
for(i=l2; i<=r2; i++)
{
if(pre[l1]==in[i])//从中序中找到当前的根节点
{
level[N]=id;
post[N++]=in[i];
len1=i-l2;//左子树的长度
len2=r2-i;//右子树的长度
break;
}
}
postTranverse(r1-len2+1,r1,i+1,r2,id+1);//递归右子树
postTranverse(l1+1,l1+len1,l2,i-1,id+1);//递归左子树 }
int main(int argc, char** argv)
{
int n,i,j;
cin>>n;
for(i=0; i<n; i++)
{
scanf("%d",&in[i]);//中序遍历序列
}
for(i=0; i<n; i++)
{
scanf("%d",&pre[i]);//前序遍历序列
}
memset(level,-1,sizeof(level));
postTranverse(0,n-1,0,n-1,0);
int t=1;
printf("%d",post[0]);
for(i=1; i<n; i++)
{
for(j=1; j<n; j++)
{
if(level[j]==t)
{
printf(" %d",post[j]);
}
}
++t;
}
return 0;
}