【leetcode hot 100 105】从前序与中序遍历序列构造二叉树

错误解法一:preorder[0]为根节点,在inorder中找到preorder[0]的位置numInorder,其左边为左子树,右边为右子树。利用Arrays.copyOfRange()函数来取数组子集。

java 复制代码
/**
 * Definition for a binary tree node.
 * public class TreeNode {
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode() {}
 *     TreeNode(int val) { this.val = val; }
 *     TreeNode(int val, TreeNode left, TreeNode right) {
 *         this.val = val;
 *         this.left = left;
 *         this.right = right;
 *     }
 * }
 */
class Solution {
    public TreeNode buildTree(int[] preorder, int[] inorder) {
        if(preorder.length==0){
            return null;
        }
        // 找到preorder[0]在inoder中的位置
        int numInoger;
        for(numInoger=0;numInoger<inorder.length;numInoger++){
            if(preorder[0]==inorder[numInoger]){
                break;
            }
        }
        int[] newPreorder = Arrays.copyOfRange(preorder,1,preorder.length);
        int[] newInoderLeft = Arrays.copyOfRange(inorder,0,numInoger);
        int[] newInoderRight = Arrays.copyOfRange(inorder,numInoger+1,inorder.length);
        TreeNode left = buildTree(newPreorder, newInoderLeft);
        TreeNode right = buildTree(newPreorder, newInoderRight);

        return new TreeNode(preorder[0], left, right);
    }
}

错误原因:

int[] newInoderRight = Arrays.copyOfRange(inorder,numInoger+1,inorder.length)Arrays.copyOfRange()无法处理numInoger+1==inorder.length的情况。

解法一:由解法二发现的问题改进上述错误解法 =》速度慢!!!!

java 复制代码
/**
 * Definition for a binary tree node.
 * public class TreeNode {
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode() {}
 *     TreeNode(int val) { this.val = val; }
 *     TreeNode(int val, TreeNode left, TreeNode right) {
 *         this.val = val;
 *         this.left = left;
 *         this.right = right;
 *     }
 * }
 */
class Solution {
    public TreeNode buildTree(int[] preorder, int[] inorder) {
        if(preorder.length==0){
            return null;
        }
        // 找到preorder[0]在inoder中的位置
        int numInoger;
        for(numInoger=0;numInoger<inorder.length;numInoger++){
            if(preorder[0]==inorder[numInoger]){
                break;
            }
        }
        // 得到左子树中的节点数目
        int num_left = numInoger;
        int[] newPreorderLeft = Arrays.copyOfRange(preorder,1,1+numInoger);
        int[] newPreorderRight = Arrays.copyOfRange(preorder,1+numInoger,preorder.length);
        int[] newInoderLeft = Arrays.copyOfRange(inorder,0,numInoger);
        int[] newInoderRight = Arrays.copyOfRange(inorder,numInoger+1,inorder.length);
        TreeNode left = buildTree(newPreorderLeft, newInoderLeft);
        TreeNode right = buildTree(newPreorderRight, newInoderRight);

        return new TreeNode(preorder[0], left, right);
    }
}

解法二:preorder[0]为根节点,在inorder中找到preorder[0]的位置numInorder,其左边为左子树,右边为右子树。利用双指针函数来取数组子集。

java 复制代码
/**
 * Definition for a binary tree node.
 * public class TreeNode {
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode() {}
 *     TreeNode(int val) { this.val = val; }
 *     TreeNode(int val, TreeNode left, TreeNode right) {
 *         this.val = val;
 *         this.left = left;
 *         this.right = right;
 *     }
 * }
 */
class Solution {
    public TreeNode buildTree(int[] preorder, int[] inorder) {
        int n = preorder.length;
        return myBuildTree(preorder, inorder, 0, n - 1, 0, n - 1);
    }

    public TreeNode myBuildTree(int[] preorder, int[] inorder, int preorder_left, int preorder_right, int inorder_left, int inorder_right){
        if(preorder_left>preorder_right){
            return null;
        }
        // 找到preorder[preorder_left]在inoder中的位置
        int numInoger;
        for(numInoger=inorder_left;numInoger<=inorder_right;numInoger++){
            if(preorder[preorder_left]==inorder[numInoger]){
                break;
            }
        }
        
        TreeNode root = new TreeNode(preorder[preorder_left]);
        // 得到左子树中的节点数目
        int num_left = numInoger - inorder_left;
        root.left = myBuildTree(preorder, inorder, preorder_left+1, preorder_left+num_left, inorder_left, numInoger-1);
        root.right = myBuildTree(preorder, inorder,preorder_left+num_left+1, preorder_right, numInoger+1, inorder_right);
        return root;
    }
}

注意:

  • 不能使用TreeNode left = myBuildTree(preorder, inorder, preorder_left+1, preorder_right, inorder_left, numInoger-1)TreeNode right = myBuildTree(preorder, inorder, preorder_left+1, preorder_right, numInoger+1, inorder_right),递归的不是preorder余下的所有数,而是要区分余下的数哪些在左子树+哪些在右子数:利用左子树中的节点数目num_left = numInoger - inorder_left来确定
  • 速度慢 =》改进:构造哈希映射,帮助我们快速定位根节点HashMap(inorder[i], i)
相关推荐
爱编程的化学家7 分钟前
代码随想录算法训练营第27天 -- 动态规划1 || 509.斐波那契数列 / 70.爬楼梯 / 746.使用最小花费爬楼梯
数据结构·c++·算法·leetcode·动态规划·代码随想录
CoovallyAIHub18 分钟前
告别等待!十条高效PyTorch数据增强流水线,让你的GPU不再"饥饿"
深度学习·算法·计算机视觉
海琴烟Sunshine36 分钟前
leetcode 66.加一 python
python·算法·leetcode
rengang661 小时前
09-随机森林:介绍集成学习中通过多决策树提升性能的算法
人工智能·算法·随机森林·机器学习·集成学习
CoovallyAIHub1 小时前
量子计算迎来诺奖时刻!谷歌赢麻了
深度学习·算法·计算机视觉
法拉第第2 小时前
caffine概率统计算法之Count-Min Sketch
算法
法拉第第2 小时前
淘汰策略之tinyLFU
算法
mit6.8242 小时前
[Tongyi] 工具集成 | run_react_infer
人工智能·深度学习·算法
闻缺陷则喜何志丹2 小时前
【C++贪心】P8769 [蓝桥杯 2021 国 C] 巧克力|普及+
c++·算法·蓝桥杯·洛谷