PTA 7-2 树的遍历 (25 分)

7-2 树的遍历 (25 分)

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

输入格式：
输入第一行给出一个正整数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
结尾无空行

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

class Node {
public:
    int value;
    Node *leftNode;
    Node *rightNode;

    Node(int value) : value(value) {}
};

Node *createTree(int pF, int pS, int iF, int iS);
void breathOrder(Node *root);
int nodeNum;// 节点个数
int *post;// 后序遍历
int *inorder;// 中序遍历
int main() {
    cin >> nodeNum;
    inorder = new int[nodeNum];
    post = new int[nodeNum];
    for (int i = 0; i < nodeNum; ++i) {
        cin >> post[i];
    }
    for (int i = 0; i < nodeNum; ++i) {
        cin >> inorder[i];
    }
    // 构建二叉树
    Node *root = createTree(0, nodeNum - 1, 0, nodeNum - 1);
    breathOrder(root);// 横向遍历二叉树
    delete[] post;
    delete[] inorder;
    return 0;
}

void breathOrder(Node *root) {
    if (root == nullptr) {
        return;
    }
    queue<Node *> deque;
    deque.push(root);
    int n = 0;
    while (!deque.empty()) {
        Node *popNode = deque.front();
        deque.pop();
        cout << popNode->value;
        if (n++ != nodeNum - 1) {
            cout << " ";
        }
        if (popNode->leftNode != nullptr) {
            deque.push(popNode->leftNode);
        }
        if (popNode->rightNode != nullptr) {
            deque.push(popNode->rightNode);
        }
    }
}

/**
 * @param pF  后序遍历起始节点
 * @param pS 后序遍历结束节点
 * @param iF 中序遍历起始节点
 * @param iS 中序遍历结束节点
 * @return 子树的根节点
 */
Node *createTree(int pF, int pS, int iF, int iS) {
    if (pF == pS) {// 该判断可有可无
        return new Node(post[pF]);
    }
    Node *header = new Node(post[pS]);
    // 计算左子树节点个数
    int leftNum = 0;
    while (inorder[leftNum + iF] != header->value) {
        ++leftNum;
    }
    // 构建左子树
    if (leftNum > 0)
        header->leftNode = createTree(pF, pF + leftNum - 1, iF, iF + leftNum - 1);
    // 构建右子树
    if (leftNum < pS - pF)
        header->rightNode = createTree(pF + leftNum, pS - 1, iF + leftNum + 1, iS);
    return header;
}