力扣刷题（剑指 Offer 07. 重建二叉树）

剑指 Offer 07. 重建二叉树 - 力扣（LeetCode）

输入某二叉树的前序遍历和中序遍历的结果，请构建该二叉树并返回其根节点。

假设输入的前序遍历和中序遍历的结果中都不含重复的数字。

示例 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]

思路：重建二叉树，总体上的思路就说按照前序的第一个一定是该子树的根节点，然后再去寻找这个节点在中序中的位置，中序遍历结果中，这个位置之前就说左子树的内容，位置之后就是右子树的内容。调用的迭代函数所有需要的参数如下：

用一个hashmap来存放二叉树的每一个节点在中序遍历中的位置；该子树在前序中的起始位置和终止位置，该子树在中序中的起始位置。

注：通过上述的逻辑分析可知，这一套逻辑只能用于没用重复节点的情况下，一旦题目中表明可以有重复的节点，这个方法就不能使用了。

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode(int x) : val(x), left(NULL), right(NULL) {}
 * };
 */
class Solution {
public:
    TreeNode* buildTree(vector<int>& preorder, vector<int>& inorder) {
        if(preorder.empty())return NULL;
        unordered_map<int,int> loc_inorder;
        int n=inorder.size();
        for(int i=0;i<n;++i){//存放每个节点在中序中的位置
            loc_inorder[inorder[i]]=i;
        }
        return build(preorder,loc_inorder,0,n-1,0);
    }
private:
    TreeNode* build(vector<int>& preorder,unordered_map<int,int>& loc_inorder,int ll_pre,int rr_pre, int ll_in){
        //ll_pre 和 rr_pre 是前序遍历下的左右端点位置，ll_in表示的是中序中的起始位置
        //创建根节点，根节点就说前序遍历的第一个
        TreeNode *node=new TreeNode(preorder[ll_pre]);
        if(ll_pre==rr_pre)return node;
        //分别确定左右子树创建的前序遍历的范围
        //loc表示当前的根节点在中序遍历中的位置
        int loc=loc_inorder[preorder[ll_pre]],left_len=loc-ll_in;
        if(left_len>0)node->left=build(preorder,loc_inorder,ll_pre+1,ll_pre+left_len,ll_in);
        if(rr_pre-ll_pre-left_len>0)node->right=build(preorder,loc_inorder,ll_pre+left_len+1,rr_pre,loc+1);
        return node;
    }
};