剑指offer九:重建二叉树
#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->value = root_value;
root->left = NULL;
root->right = NULL;
cout << root_value << " ";
if (preorder == endpreorder && inorder == endinorder)
return root;
int* rootIndex = inorder;
while (*rootIndex != root_value && rootIndex < endpreorder)
rootIndex++;
int left_len = rootIndex - inorder;
int* left_preorder_end = preorder + left_len;
//构建左子树
if (left_len > 0)//左子树只要大于0就要构建右子树
{
root->left = construct_method(preorder + 1, left_preorder_end, inorder, rootIndex - 1);
}
//构建右子树
if (endpreorder - preorder - left_len > 0)//右子树只要大于0就要构建右子树
{
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);
}