PTA玩转二叉树

Posted 2021-03-16 andre

L2-011 玩转二叉树 (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 <stdio.h>
#include <stdlib.h>
typedef struct nold
{
    int data;
    struct nold *lchild,*rchild;
} Bitnold,*Bitree;//定义结构体
void creat(int prer[],int i,int j,int mid[],int k,int h,Bitnold **t)//由前序遍历序列和中序遍历序列恢复二叉树
{
    *t=(Bitnold*)malloc(sizeof(Bitnold));
    (*t)->data=prer[i];
    int m=k;
    while(prer[i]!=mid[m])
        m++;
    if(m==k)
        (*t)->lchild=NULL;
    else
        creat(prer,i+1,i+m-k,mid,k,m-1,&(*t)->lchild);
    if(m==h)
        (*t)->rchild=NULL;
    else
        creat(prer,i+m-k+1,j,mid,m+1,h,&(*t)->rchild);
}

int main()
{
    int i;
    Bitnold *root;
    int n;
    scanf("%d",&n);
    int mid[n];
    int prer[n];
    for(i=0; i<n; i++)
        scanf("%d",&mid[i]);
    for(i=0; i<n; i++)
        scanf("%d",&prer[i]);
    creat(prer,0,n-1,mid,0,n-1,&root);
    change(root);
    LevelOrder(root);
    return 0;
}

void change(Bitnold *r) //左右子树交换
{
    Bitnold *temp=NULL;
    if(r)
    {
        if(r==NULL) return;
        temp = r->lchild;
        r->lchild=r->rchild;
        r->rchild = temp;
        change(r->lchild);
        change(r->rchild);
    }
}

void PreOrder(Bitnold *r)//前序遍历
{
    if(r)
    {
        printf("%d ",r->data);
        PreOrder(r->lchild);
        PreOrder(r->rchild);
    }
}

void LevelOrder(Bitnold *r) //层次遍历
{
    Bitnold *queue[100];
    int front,rear,i=1,j=0;
    int arr[100];
    if(r==NULL) return;
    front=-1;
    rear=0;
    queue[rear]=r;
    while(front!=rear)
    {
        front++;
        arr[j]=queue[front]->data;
        j++;
        if(queue[front]->lchild!=NULL)
        {
            rear++;
            queue[rear]=queue[front]->lchild;
        }
        if(queue[front]->rchild!=NULL)
        {
            rear++;
            queue[rear]=queue[front]->rchild;
        }
    }
    printf("%d",arr[0]);
    for(i=1;i<j;i++)
    {
        printf(" %d",arr[i]);
    }
    return 0;
}