OpenJudge 由中根順序和后根序列重建二叉樹

題目內容：

我們知道如何按照三種深度優先次序來周游一棵二叉樹，來得到中根序列、前根序列和后根序列。反過來，如果給定二叉樹的中根序列和后根序列，或者給定中根序列和前根序列，可以重建一二叉樹。本題輸入一棵二叉樹的中根序列和后根序列，要求在內存中重建二叉樹，最后輸出這棵二叉樹的前根序列。

用不同的整數來唯一標識二叉樹的每一個結點，下面的二叉樹

 

中根序列是9 5 32 67

后根序列9 32 67 5

前根序列5 9 67 32

 

輸入格式:

兩行。第一行是二叉樹的中根序列，第二行是后根序列。每個數字表示的結點之間用空格隔開。結點數字范圍0～65535。暫不必考慮不合理的輸入數據。

 

輸出格式：

一行。由輸入中的中根序列和后根序列重建的二叉樹的前根序列。每個數字表示的結點之間用空格隔開。

 

 

【題解】

        后根序列的最后一個元素即為二叉樹的樹根root。root將中根序列分為兩部分，左半邊是左子樹的中根序列，而右半邊則是右部分的中根序列。同時后根序列依照左子樹和右子樹節點數也可以被分為左子樹的后根序列和右子樹的后根序列。於是便可依此遞歸地按左右子樹的后根、中根序列重建子樹，最終重建二叉樹。

【代碼】

#include <iostream>
using namespace std;
/*build BT from its inorder and postorder*/


constexpr int MAXN = 65540;
int postorder[MAXN], inorder[MAXN];

struct btn
{
    int data;
    btn *left;
    btn *right;
};

btn *build_tree(int io1, int io2, int po1, int po2)
{
    /*io1, io2 are the begining and ending points of inorder sequence respectively*/
    /*po1, po2 are the begining and ending points of postorder sequence respectively*/
    int i = 0, ion = io2 - io1 + 1;
    btn* root = new btn;
    root->data = postorder[po2];
    root->left = NULL;
    root->right = NULL;
    for (i = 0; i < ion; i++) {
        if (root->data == inorder[io1 + i])break;
    }
    if (i > 0) root->left = build_tree(io1, io1 + i - 1, po1, po1 + i - 1);
    if (io1 + i < io2) root->right = build_tree(io1 + i + 1, io2, po1 + i, po2 - 1);
    return root;
}

void preorder(btn *root, int vnum, int &count)
{
    if (root != NULL) {
        cout << root->data;
        if (count < vnum - 1) {
            cout << " ";
            count++;
            preorder(root->left, vnum, count);
            preorder(root->right, vnum, count);
        }
        else cout << endl;
    }
}

void deletetree(btn *root)
{
    if (root != NULL) {
        delete(root->left);
        delete(root->right);
        delete root;
        root = NULL;
    }
}

int main()
{
    int c = 0, i = 0, vnum = 0;
    while (cin >> inorder[i++]) {
        if (cin.get() != ' ')break;
        /*Although cin will skip the blank space, the cursor stays at the blank after cin reads a number，thus cin.get() can fetch the 
        blank space.*/
    }

    vnum = i;
    i = 0;
    while (cin >> postorder[i++]) {
        if (cin.get() != ' ')break;
    }
    btn *root = build_tree(0, i - 1, 0, i - 1);
    preorder(root, vnum, c);
    deletetree(root);
    return 0;
}