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

Posted 2022-05-26 放羊的牧码

题目链接：点击打开链接

题目大意：略

解题思路：略

相关企业

• 字节跳动
• Facebook
• 亚马逊（Amazon）
• 谷歌（Google）
• 微软（Microsoft）
• 优步（Uber）
• 彭博（Bloomberg）

AC 代码

• Java

/**
 * Definition for a binary tree node.
 * public class TreeNode 
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode(int x)  val = x; 
 * 
 */
 
// 解决方案(1)
class Solution 
 
    int[] preorder, inorder;
 
    Map<Integer, Integer> inorderMap = new HashMap<>();
 
    public TreeNode buildTree(int[] preorder, int[] inorder) 
        this.preorder = preorder;
        this.inorder = inorder;
        for (int i = 0; i < inorder.length; i++) 
            inorderMap.put(inorder[i], i);
        
        return recur(0, inorder.length - 1, 0, preorder.length - 1);
    
 
    TreeNode recur(int la, int ra, int lb, int rb) 
        if (la > ra) return null;
        int preRoot = preorder[lb];
        int midRoot = inorderMap.get(preRoot);
        int leftLen = midRoot - la;
        TreeNode root = new TreeNode(preRoot);
        root.left = recur(la, midRoot - 1, lb + 1, lb + leftLen);
        root.right = recur(midRoot + 1, ra, lb + leftLen + 1, rb);
 
        return root;
    

 
// 解决方案(2)
class Solution 
    int[] preorder;
    HashMap<Integer, Integer> dic = new HashMap<>();
    public TreeNode buildTree(int[] preorder, int[] inorder) 
        this.preorder = preorder;
        for(int i = 0; i < inorder.length; i++)
            dic.put(inorder[i], i);
        return recur(0, 0, inorder.length - 1);
    
    TreeNode recur(int root, int left, int right) 
        if(left > right) return null;                          // 递归终止
        TreeNode node = new TreeNode(preorder[root]);          // 建立根节点
        int i = dic.get(preorder[root]);                       // 划分根节点、左子树、右子树
        node.left = recur(root + 1, left, i - 1);              // 开启左子树递归
        node.right = recur(root + i - left + 1, i + 1, right); // 开启右子树递归, root + i - left + 1 => root + (left - (i - 1) + 1) + 1 (根节点索引 + 左子树长度 + 1)
        return node;                                           // 回溯返回根节点
    

• C++

/**
 * 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) 
        this->preorder = preorder;
        for(int i = 0; i < inorder.size(); i++)
            dic[inorder[i]] = i;
        return recur(0, 0, inorder.size() - 1);
    
private:
    vector<int> preorder;
    unordered_map<int, int> dic;
    TreeNode* recur(int root, int left, int right)  
        if(left > right) return nullptr;                        // 递归终止
        TreeNode* node = new TreeNode(preorder[root]);          // 建立根节点
        int i = dic[preorder[root]];                            // 划分根节点、左子树、右子树
        node->left = recur(root + 1, left, i - 1);              // 开启左子树递归
        node->right = recur(root + i - left + 1, i + 1, right); // 开启右子树递归, root + i - left + 1 => root + (left - (i - 1) + 1) + 1 (根节点索引 + 左子树长度 + 1)
        return node;                                            // 回溯返回根节点
    
;