PTA L2-011玩转二叉树

题目描述

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

• 输入格式：

  输入第一行给出一个正整数 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

解题思路：

• 刚开始受L2-006的影响，想要把它的数组形式的层序遍历结果展现出来，再记录好结点的父母结点，然后再构造一棵左右结点颠倒的树，但是发现这样做太繁琐了，而且非常混乱，直接另辟蹊径。

• 有了前序（或者后序），再加上中序遍历其实就可以构造一棵树了，这个之前我是不知道的，主要思想就是利用递归，递归解决的就是从整棵树到子树的构建过程，拿这题举例子。

  • 前序+中序：前序最左边的就是根结点，且左边部分为左子树，右边的部分为右子树；对于中序，需要先从前序找到根结点，然后找到中序的位置，就可以区分出左右子树，中序遍历左边肯定是左子树，右边是右子树，根据这个思想，每次找出子树的结点个数len，加上前序的起点下标，以及中序的起点下标，进行递归计算；到结点个数为0时，说明你这个结点的左子树没有结点，直接返回NULL，对于结点个数不为0的那些根结点则需要根据递归结束返回T才能知道它的左右结点的内容。

  • 需要非常熟悉前序和中序遍历的特性，从而根据这个构造出树。有了前序起点和中序起点以及len长度，就可以知道这个子树所包含的结点；前序结点的起始位置对应的点就是该子树的根结点，随后就需要根据中序的起点找到中序遍历序列中值等于根结点的下标，从而可以将子树又分成两棵子树，也可以分别计算出左右子树分别的结点个数。

  • 左子树的中序起始点是与上次遍历的一样，而前序结点需要 [原来+1]，因为前序遍历左边的是左子树的部分，而刚刚的前序结点用过了，所以 [原来+1]，继续找这个根结点对应的左右子树，长度就是左子树的长度。

  • 右子树的中序结点起始是 [原来+len+1]，在前序结点起始也是 [原来+len+1]，对于长度 [原来-len-1]。

  • 关键就是找子树根结点的中序遍历的位置，算出左、右子树的结点数，计算好左、右子树的中、前序的在整个序列中的起始点。

  • 因为需要镜面，只要再构建树的时候把左右子树的顺序交换一下即可。

代码：

#include <iostream>
#include <queue>
using namespace std;
typedef struct BiTNode
{
    int data;
    BiTNode *lchild, *rchild;
} BiTNode, *BiTree;
int mid[35];
int front[35];
int main()
{
    void Leve_Order(BiTree T);
    BiTree constructATree(int midP, int frontP, int len);
    // BiTree Create(int x, int y, int z); //中序起点 前序起点 长度

    int N;
    cin >> N;
    int i;
    BiTNode *T;
    for (i = 1; i <= N; i++)
    {
        cin >> mid[i];
    }

    for (i = 1; i <= N; i++)
    {
        cin >> front[i];
    }
    T = constructATree(1, 1, N);
    Leve_Order(T);
    return 0;
}

BiTree constructATree(int midP, int frontP, int len)
{
    if (len == 0)
    {
        return NULL;
    }
    int i;
    int root = 0;
    int newLen = 0;
    BiTNode *T = new BiTNode;
    root = front[frontP];
    T->data = root;

    for (i = midP; i < midP + len; i++)
    {
        if (mid[i] == root)
        {
            newLen = i - midP;
            break;
        }
    }
    T->rchild = constructATree(midP, frontP + 1, newLen);
    T->lchild = constructATree(midP + newLen + 1, frontP + newLen + 1, len - newLen - 1);
    return T;
}


bool flag = 0;

//层序遍历
void Leve_Order(BiTree T)
{
    queue<BiTNode*>q;
    q.push(T);
    while(!q.empty())
    {
        BiTNode *p = q.front();
        q.pop();
        if(flag) cout<<' ';
        cout<<p->data;
        flag = 1;
        if(p->lchild)
            q.push(p->lchild);
        if(p->rchild)
            q.push(p->rchild);
    }
}