劍指offer 面試題6：重建二叉樹

重建二叉樹

題目

　　輸入某二叉樹的前序遍歷和中序遍歷，請重建出該二叉樹。假設輸入的前序遍歷和中序遍歷的結果中都不含有重復的數字。

　　例如，前序遍歷序列：｛1，2，3，7，3，5，6，8｝，中序遍歷序列：｛4，7，2，1，5，3，8，6｝

答案

　　前序遍歷：

　　　　前序遍歷首先訪問根結點然后遍歷左子樹，最后遍歷右子樹。在遍歷左、右子樹時，仍然先訪問根結點，然后遍歷左子樹，最后遍歷右子樹。

　　中序遍歷：

　　　　中序遍歷首先遍歷左子樹，然后訪問根結點，最后遍歷右子樹。在遍歷左、右子樹時，仍然先遍歷左子樹，再訪問根結點，最后遍歷右子樹。

#include <iostream>

using namespace std;

struct binary_tree_node{
    int value;
    binary_tree_node*  left;
    binary_tree_node*  right;
};

binary_tree_node* binary_tree_constuct(int* preorder, int* inorder, int length);

int main()
{
    int pre[8] = {1,2,4,7,3,5,6,8};
    int in[8]  = {4,7,2,1,5,3,8,6};

    binary_tree_node* root = binary_tree_constuct(pre, in, 8);
}

binary_tree_node* construct_method(int* preorder, int* endpreorder, int* inorder, int* endinorder)
{
    int root_value = preorder[0];
    binary_tree_node* root = new binary_tree_node();
    root->left = NULL;
    root->right = NULL;

    cout<<root_value<<" ";

    if(preorder == endpreorder && inorder == endinorder)
        return root;

    int* rootIndex = preorder;
    rootIndex++;
    while(*rootIndex != root_value && rootIndex < endpreorder)
        rootIndex++;

    int left_len = rootIndex - preorder;
    int* left_preorder_end = preorder + left_len;
    //left
    if(left_len > 0)
    {
        root->left = construct_method(preorder+1, left_preorder_end, inorder, rootIndex-1);
    }
    //right
    if(left_len < endpreorder - preorder)
    {
        root->right = construct_method(left_preorder_end+1, endpreorder, rootIndex+1, endinorder);
    }

    return root;
}

binary_tree_node* binary_tree_constuct(int* preorder, int* inorder, int length)
{
    if(preorder == NULL || inorder == NULL || length <= 0)
    {
        return NULL;
    }

    return construct_method(preorder, preorder+length-1, inorder,inorder+length-1);

}