剑指Offer：[第20天 分治算法(中等)]---＞重建二叉树

---

一、题目描述

输入某二叉树的前序遍历和中序遍历的结果，请构建该二叉树并返回其根节点。假设输入的前序遍历和中序遍历的结果中都不含重复的数字。

示例1：

Input: preorder = [3,9,20,15,7], inorder = [9,3,15,20,7]
Output: [3,9,20,null,null,15,7]

示例2：

Input: preorder = [-1], inorder = [-1]
Output: [-1]

限制：

0 <= 节点个数 <= 5000

二、思路分析

注：思路分析中的一些内容和图片参考自力扣各位前辈的题解，感谢他们的无私奉献

思路

前序遍历性质： 节点按照 [ 根节点 | 左子树 | 右子树 ] 排序
中序遍历性质： 节点按照 [ 左子树 | 根节点 | 右子树 ] 排序
根据以上性质，可得出以下推论：
①前序遍历的首元素为树的根节点node的值----可以确定根节点
②在中序遍历中搜索根节点node的索引，可将中序遍历划分为 [ 左子树 | 根节点 | 右子树 ] ----可以确定左右子树的节点数
③根据中序遍历中的左(右)子树的节点数量，可将前序遍历划分为 [ 根节点 | 左子树 | 右子树 ] ----根据左右子树的节点数，可以确定出左右子树的根结点

根据分治算法思想，对于树的左、右子树，仍可复用以上方法划分子树的左右子树。
分治算法流程：
①递推参数：根节点在前序遍历的索引root、子树在中序遍历的左边界left、子树在中序遍历的右边界right
②终止条件：当left>right，代表已经越过叶节点，此时返回null
③递推工作：
----建立根节点node：节点值为preorder[root]
----划分左右子树：查找根节点在中序遍历inorder中的索引i。(为了提升效率，可以考虑使用哈希表dic存储中序遍历的值与索引的映射，查找操作的时间复杂度为O(1)。但是本题我们使用C语言实现，且二叉树中各个节点值的范围未确定，因此我们仍然使用循环来查找根节点位置)
----构建左右子树：开启左右子树递归

④返回值：回溯返回node，作为上一层递归中根节点的左/右子节点
案例分析：











复杂度分析：
时间复杂度$\mathrm{O}(\mathrm{N}\mathrm{l}\mathrm{o}\mathrm{g}\mathrm{N})$：其中N为树的节点数量。递归共建立N个节点，每层递归中的节点建立操作占用O(1)，搜索根结点操作平均占O(logN)，因此使用O(NlogN)时间。
空间复杂度$\mathrm{O}(\mathrm{N})$：最差情况下(输入二叉树为链表时)，递归深度达到N，占用O(N)的栈帧空间，

三、整体代码

整体代码如下：

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     struct TreeNode *left;
 *     struct TreeNode *right;
 * };
 */

struct TreeNode* recur(int* preorder, int* inorder, int root, int left, int right){
    if(left > right) return NULL;
    struct TreeNode* node = (struct TreeNode*)malloc(sizeof(struct TreeNode));
    int j;
    for(j = 0; j < right; j++){
        if(inorder[j] == preorder[root]) break;
    }
    int i = j;
    node->val = preorder[root];
    node->left = recur(preorder, inorder, root+1, left, i-1);
    node->right = recur(preorder, inorder, root+i-left+1, i+1, right);
    return node;
}

struct TreeNode* buildTree(int* preorder, int preorderSize, int* inorder, int inorderSize){
    return recur(preorder, inorder, 0, 0, inorderSize-1);
}

运行，测试通过