玩转二叉树

7-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;

struct treeNode{
    int Data;
    struct treeNode *right;
    struct treeNode *left;
};
typedef struct treeNode* Node;

Node createTree(int *X,int *Z,int temp){
    Node root=(Node)malloc(sizeof(struct treeNode));
    if(temp==0)return NULL;
    root->left=NULL;
    root->right=NULL;
    root->Data=*X;
    int k;
    for(int i=0;i<temp;i++){
        if(*X==*(Z+i))
            {k=i;break;}
    }
    root->left=createTree(X+1,Z,k);
    root->right=createTree(X+k+1,Z+k+1,temp-k-1);
    return root;
}

void fanzhuan(Node root)
{
    if(root==NULL)return;
    Node key;
    if(root->left!=NULL||root->right!=NULL)
    {
        key=root->left;
        root->left=root->right;
        root->right=key;
    }
    fanzhuan(root->left);
    fanzhuan(root->right);
}

void LevelorderTraversal( Node BT )
{
    if(BT==NULL)return;
    Node CC[10000];
    CC[0]=BT;
    int lens=1;
    int hhh=0;
    while(1)
    {
        if(lens==0)return;
        int temp=0;Node TH[10000];
        int k=0;
        for(int i=0;i<lens;i++)
        {
            if(CC[i]!=NULL)
                {
                    if(hhh==0)
                    printf("%d",CC[i]->Data);
                    else
                        printf(" %d",CC[i]->Data);
                }
            if(CC[i]->left!=NULL)
                TH[temp++]=CC[i]->left;
            if(CC[i]->right!=NULL)
                TH[temp++]=CC[i]->right;
        }
        lens=temp;
        for(int i=0;i<lens;i++)
            CC[i]=TH[i];
        hhh++;
    }
}

int main()
{
    int N;
    cin>>N;
    int xx[N];
    int zx[N];
    for(int i=0;i<N;i++)
        cin>>xx[i];
    for(int i=0;i<N;i++)
        cin>>zx[i];
    Node Root=createTree(zx,xx,N);
    fanzhuan(Root);
    LevelorderTraversal(Root);
    return 0;
}

猜你喜欢