PTA 根据后序和中序遍历输出层序遍历

给定一棵二叉树的后序遍历和中序遍历，请你输出其层序遍历的序列。这里假设键值都是互不相等的正整数。

输入格式：

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

输出格式：

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

输入样例：

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

输出样例：

4 1 6 3 5 7 2

 思路：1.根据中序遍历和后序遍历特点构建二叉树

            2.层序遍历

可以发现后序遍历的数组最后一个元素代表的即为根节点。知道这个性质后，我们可以利用已知的根节点信息在中序遍历的数组中找到根节点所在的下标，然后根据其将中序遍历的数组分成左右两部分，左边部分即左子树，右边部分为右子树，针对每个部分可以用同样的方法继续递归下去构造。

#include<iostream>
#include<queue>
using namespace std;

typedef int DataType;

typedef struct TreeNode {
    DataType val;
    struct TreeNode* left;
    struct TreeNode* right;
    TreeNode(DataType data):val(data), left(NULL), right(NULL) {};
}TNode;

//构建二叉树
TNode* Create_Tree(int n,int* inorder,int* postorder) {
    if (n < 0) return NULL;//n<0 返回上一级函数

    TNode* root = new TreeNode(postorder[n]);
    int i = 0;
    for (i; i < n; i++) {
        if (inorder[i] == postorder[n]) break;
    }
    root->left = Create_Tree(i - 1, inorder, postorder);
    root->right = Create_Tree(n-i-1, inorder+i+1, postorder+i);
    return root;
}

//层序遍历
void LevelOrder(TNode* root) {
    queue<TNode*> Q;
    if(root) Q.push(root);
    while (!Q.empty()) {
        TNode* t = Q.front();
        printf("%d ", t->val);
        Q.pop();
        if (t->left) Q.push(t->left);
        if (t->right) Q.push(t->right);
    }

}

int main() {

    int n;
    cin >> n;
    int* inorder = (int*)malloc(sizeof(int) * n);//中序
    int* postorder = (int*)malloc(sizeof(int) * n);//后序
    for (int i = 0; i < n; i++)
    {
        cin >> postorder[i];
    }
    for (int i = 0; i < n; i++)
    {
        cin >> inorder[i];
    }
    TNode* root = Create_Tree(n - 1, inorder, postorder);
    LevelOrder(root);

	return 0;
}