JZ7 重建二叉树

题目描述

代码：

struct TreeNode* construct_core(int* pre_order, int pre_len, int* in_order, int in_len){
    struct TreeNode* root = (struct TreeNode*)malloc(sizeof(struct TreeNode));
    int left = 0, right = 0, i=0;
    root->val = pre_order[0];
    root->left = NULL;
    root->right = NULL;
    if (pre_len == 0 && in_len == 0){
        return NULL;
    }
    for(i=0;i<pre_len;i++){
        if(pre_order[0] == in_order[i]){
            left = i;
            right = pre_len - left - 1;
        }
    }
    if(left >= 0){
        root->left = construct_core(pre_order+1, left, in_order, left);
    }
    if(right >= 0){
        root->right = construct_core(pre_order+left+1, right, in_order+left+1, right);
    }
    return root;
}
struct TreeNode* reConstructBinaryTree(int* pre, int preLen, int* vin, int vinLen ) {
    return construct_core(pre, preLen, vin, vinLen);
}

        通过观察发现，前序遍历的第一个元素是根节点，在中序遍历的数组中寻找pre[0]，则在中序左侧的全为根节点的左侧，右侧则为根节点的右侧，这样就可以知道根节点左右两侧分别有多少个节点。

        然后，先解决左边，可以发现左侧可以拆分成一个小的根节点点，此时就意识到可以使用迭代，当根节点的下方全为空时，迭代结束。迭代的收敛条件为传入的数组逐渐减小，当数组的前序遍历和中序遍历的长度都为0时，则返回NULL，表面当前根节点不存在。右侧也是如此。