C4top-玩转二叉树（根据前序中序镜像反转后层次遍历）

玩转二叉树  

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

输入格式：

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

输出格式：

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

输入样例：

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

输出样例：

4 6 1 7 5 3 2

 

• 时间限制：400ms
• 内存限制：64MB
• 代码长度限制：16kB
• 判题程序：系统默认
• 作者：陈越
• 单位：浙江大学

题目判定

解题程序

#include<cstdio>
#include<cstring>
#include<cmath>
#include<iomanip>
#include<cstdlib>
#include<iostream>
#include<algorithm>
using namespace std;
#define MAXN 40
#define INF 0x3f3f3f3f
typedef struct node//二叉树
{
    int data;
    struct node * lchild;
    struct node * rchild;
} btnode;
int n;
int pre[MAXN],in[MAXN];

btnode *creat(int * pre, int * in,int n)//前序，中序，节点数
{
    btnode * b;
    int * p;
    int k;
    if(n<=0) return NULL;
    b=(btnode*)malloc(sizeof(btnode));
    b->data=*pre;
    for(p=in; p<in+n; ++p)
        if(*p==*pre)
            break;
    k=p-in;
    b->lchild=creat(pre+1,in,k);
    b->rchild=creat(pre+k+1,p+1,n-k-1);
    return b;
}
int dis[MAXN],cnt=0;//存储遍历到的元素
void level(btnode * b,int n)//层次遍历
{
    btnode *p;
    btnode *q[n];
    int front,rear;
    front=rear=-1;
    ++rear;
    q[rear]=b;
    while(front!=rear)
    {
        front=(front+1)%n;
        p=q[front];
        dis[cnt++]=p->data;//遍历时的当前数组
        //先输出右孩子再输出左孩子来实现镜像反转
        if(p->rchild!=NULL)
        {
            rear=(rear+1)%n;
            q[rear]=p->rchild;
        }
        if(p->lchild!=NULL)
        {
            rear=(rear+1)%n;
            q[rear]=p->lchild;
        }
    }
}
int main()
{
#ifdef ONLINE_JUDGE
#else
    freopen("G:/cbx/read.txt","r",stdin);
//freopen("G:/cbx/out.txt","w",stdout);
#endif
    ios::sync_with_stdio(false);
    cin.tie(0);
    cin>>n;
    for(int i=0; i<n; ++i)
        cin>>in[i];//中序
    for(int i=0; i<n; ++i)
        cin>>pre[i];//前序
    btnode * b;
    b=creat(pre,in,n);//建树
    level(b,n);//层次遍历
    for(int i=0; i<cnt-1; ++i)
        cout<<dis[i]<<" ";
    cout<<dis[cnt-1]<<endl;
    return 0;
}