浙大PTA 玩转二叉树 (25point(s))

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

输入格式：
输入第一行给出一个正整数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<queue>
using namespace std;

#define MaxSize 30
typedef struct TNode * PtrToNode;
struct TNode
{
    int Data;
    PtrToNode Left, Right;
};
typedef PtrToNode Tree;

Tree BulidTree(int inL, int preL, int n);
void fanzhuan(Tree &T);
void LevelorderTraversal(Tree T);

int in[MaxSize] = {0}, pre[MaxSize] = {0};
int main()
{
    int n;
    scanf("%d", &n);
    for(int i = 0; i < n; i++)/*用数组in存放中序遍历*/
        scanf("%d", &in[i]);
    for(int i = 0; i < n; i++)/*用数组pre存放先序遍历*/
        scanf("%d", &pre[i]);
    Tree T;
    T = (Tree)malloc(sizeof(struct TNode));
    T = BulidTree(0, 0, n); /*建树*/
    fanzhuan(T);            /*翻转*/
    LevelorderTraversal(T); /*层序遍历*/
    return 0;
}

Tree BulidTree(int inL, int preL, int n)
{
    /*
    inL:中序遍历数组的起始位置
    preL:先序遍历数组的起始位置
    n:节点个数/数组长度
    */
    if(n == 0){return NULL;} /*长度为0, 子树为空*/
    int root, i;
    Tree T;
    /*先序遍历的第一个元素存放根节点*/
    root = pre[preL];        /*root存放根节点的值*/
    T = (Tree)malloc(sizeof(struct TNode));
    T->Data = root;
//    printf("root = %d,", root);   /*可以去掉注释,跑一下看一下运行过程*/
//    getchar();                    /*按回车,继续运行*/
    for(i = 0; i < n; i++)  /*找根节点在中序遍历的位置*/
        if(in[inL + i] == root)
            break;
    int L = i, R = n - L - 1;/*左子树的长度为L,右子树的长度为R*/
//    printf("L = %d,R = %d| ", L, R);
//    getchar();
    T->Left = BulidTree(inL, preL + 1, L);              /*调用左子树*/
    T->Right = BulidTree(inL + L + 1, preL + L + 1, R); /*调用右子树*/
    return T;
}

void fanzhuan(Tree &T)
{
    if(T == NULL)
    {
        return;
    }
    else/*交换左右子树*/
    {
        Tree Tmp;
        Tmp = T->Left;
        T->Left = T->Right;
        T->Right = Tmp;
    }
    fanzhuan(T->Left); /*对左子树进行翻转*/
    fanzhuan(T->Right);/*对右子树进行翻转*/
    return;
}
void LevelorderTraversal(Tree T)
{
    bool flag = true;
    if(!T)
        return;
    else
    {
        queue<Tree>q;
        q.push(T);
        while(!q.empty())
        {
            Tree tmp = q.front();
            if(flag){flag = false;} /*辅助格式输出*/
            else printf(" ");
            printf("%d", tmp->Data);
            q.pop();
            if(tmp->Left)
                q.push(tmp->Left);
            if(tmp->Right)
                q.push(tmp->Right);
        }
    }
}