【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)
相关推荐
Tech Synapse2 小时前
基于Surprise和Flask构建个性化电影推荐系统:从算法到全栈实现
python·算法·flask·协同过滤算法
終不似少年遊*2 小时前
国产之光DeepSeek架构理解与应用分析04
人工智能·python·深度学习·算法·大模型·ds
天天扭码2 小时前
一分钟解决 | 高频面试算法题——最大子数组之和
前端·算法·面试
杰杰批2 小时前
力扣热题100——矩阵
算法·leetcode·矩阵
明月看潮生2 小时前
青少年编程与数学 02-016 Python数据结构与算法 28课题、图像处理算法
图像处理·python·算法·青少年编程·编程与数学
_GR2 小时前
2025年蓝桥杯第十六届C&C++大学B组真题及代码
c语言·数据结构·c++·算法·贪心算法·蓝桥杯·动态规划
心想事“程”2 小时前
决策树详解+面试常见问题
算法·决策树·机器学习
照海19Gin3 小时前
数据结构中的宝藏秘籍之广义表
c语言·数据结构·算法
小oo呆4 小时前
【自然语言处理与大模型】模型压缩技术之剪枝
算法·机器学习·剪枝
bloxd yzh4 小时前
筛选法(埃氏筛法)C++
数据结构·算法