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;}