1119. Pre- and Post-order Traversals (30) 二叉树

题目地址

先序序列中的第一个以及后序序列中的最后一个（两者为同一个数）总是父结点，通过查找后序序列中最后一个结点的前一个在先序中的位置，来确定是否可以划分左右孩子，如果不能，就将其划分为右孩子（或左孩子），递归建树，中序遍历输出。

//
// Created by aleafall on 16-10-2.
//

#include <iostream>

using namespace std;

const int maxn = 31;

int n, index = 0;
int pre[maxn], post[maxn];
bool flag = true;

struct Node {
    int data;
    Node *lchild, *rchild;
} *root;

Node *create(int preL, int preR, int postL, int postR) {
    if (preL > preR) return nullptr;
    Node *node = new Node;
    node->data = pre[preL];
    node->lchild = nullptr;
    node->rchild = nullptr;
    if (preL == preR)
        return node;
    int k = 0;
    for (k = preL + 1; k <= preR; ++k) {
        if (pre[k] == post[postR - 1]) break;
    }
    if (k - preL > 1) {
        node->lchild = create(preL + 1, k - 1, postL, postL + k - preL - 2);
        node->rchild = create(k, preR, postL + k - preL - 1, postR - 1);
    } else {
        flag = false;
        node->rchild = create(k, preR, postL + k - preL - 1, postR - 1);
    }
    return node;
}

void inOrder(Node *node) {
    if (node == nullptr) return;
    inOrder(node->lchild);
    if (index < n - 1)
        cout << node->data << " ";
    else cout << node->data << endl;
    index++;
    inOrder(node->rchild);
}

int main() {
    cin >> n;
    for (int i = 0; i < n; ++i) {
        cin >> pre[i];
    }
    for (int i = 0; i < n; ++i) {
        cin >> post[i];
    }
    root = create(0, n - 1, 0, n - 1);
    if (flag) cout << "Yes\n";
    else cout << "No\n";
    inOrder(root);
    return 0;
}