L2-3 玩转二叉树 (25 分)
给定一棵二叉树的中序遍历和前序遍历,请你先将树做个镜面反转,再输出反转后的层序遍历的序列。所谓镜面反转,是指将所有非叶结点的左右孩子对换。这里假设键值都是互不相等的正整数。
输入格式:
输入第一行给出一个正整数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 <bits/stdc++.h>
using namespace std;
typedef struct BTnode
{
int data;
struct BTnode *l;
struct BTnode *r;
}BTnode;
BTnode *create(int *pre,int *in,int n)
{
BTnode *b;
b=(BTnode *)malloc(sizeof(BTnode));
int *p;
int k;
if(n<=0)
return NULL;
b->data=*pre;
for(p=in;p<in+n;p++)
if(*p==*pre)
break;
k=p-in;
b->l=create(pre+1,in,k);
b->r=create(pre+k+1,p+1,n-k-1);
return b;
}
void level(BTnode *b)
{
queue <BTnode *> qu;
BTnode *q;
qu.push(b);
int f=1;
while(!qu.empty())
{
q=qu.front();
qu.pop();
if(f==1)
{
cout<<q->data;
f=0;
}
else
cout<<" "<<q->data;
if(q->r!=NULL)
qu.push(q->r);
if(q->l!=NULL)
qu.push(q->l);
}
}
int main()
{
BTnode *B;
int i,n;
int a[51],b[51];
cin>>n;
for(i=0;i<n;i++)
{
cin>>a[i];
}
for(i=0;i<n;i++)
{
cin>>b[i];
}
B=create(b,a,n);
level(B);
return 0;
}