0
点赞
收藏
分享

微信扫一扫

LeetCode 将有序数组转换为二叉搜索树(简单、数组)day06

ZGtheGreat 2022-04-04 阅读 41
leetcode

题目描述:给你一个整数数组 nums ,其中元素已经按 升序 排列,请你将其转换为一棵 高度平衡 二叉搜索树
注意:高度平衡 二叉树是一棵满足「每个节点的左右两个子树的高度差的绝对值不超过 1 」的二叉树。
在这里插入图片描述

输入:nums = [-10,-3,0,5,9]
输出:[0,-3,9,-10,null,5]
解释:[0,-10,5,null,-3,null,9] 也将被视为正确答案:

在这里插入图片描述在这里插入图片描述

输入:nums = [1,3]
输出:[3,1]
解释:[1,null,3] 和 [3,1] 都是高度平衡二叉搜索树。

解决问题–>递归解决:题中说了要转换为一棵高度平衡的二叉搜索树,并且数组又是排过序的,这就好办了,我们可以使用递归的方式,每次取数组中间的值比如mid作为当前节点,mid前面的值作为他左子树的结点值,mid后面的值作为他右子树的节点值.
代码示例:

import java.util.Arrays;

public class ArrayToTreeNode {
    public static void main(String[] args) {

    }
    // 在原来数组的基础上再次进行调用
    public static TreeNode sortedArrayToBST1(int[] nums){
        // 递归结束的标志
        int left = 0, right = nums.length-1;
        if(left > right){
            return null;
        }

        // 找到创建当前中间节点
        int mid = (left + right) / 2;
        TreeNode root = new TreeNode(nums[mid]);

        // 构建左子树的数组
        int[] leftArray = Arrays.copyOfRange(nums, left, mid);  // 包括左边不包括右面
        // 构建右子树的数组
        int[] rightArray = Arrays.copyOfRange(nums, mid+1, nums.length);

        // 分别进行递归即可
        root.left = sortedArrayToBST(leftArray);
        root.right = sortedArrayToBST(rightArray);

        return root;
    }

	// 递归方式二, 再次调用一个函数,传入nums, left, right
    public static TreeNode sortedArrayToBST(int[] nums) {
        return dfs(nums, 0, nums.length-1);
    }

    public static TreeNode dfs(int[] nums, int left, int right){
        if(left > right){
            return null;
        }

        // 总是选择中间位置作为根节点
        int mid = (left + right) / 2;
        TreeNode root = new TreeNode(nums[mid]);
        root.left = dfs(nums, left, mid-1);
        root.right = dfs(nums, mid+1, right);

        return root;
    }
}

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;
    }
}

链接:https://leetcode-cn.com/problems/convert-sorted-array-to-binary-search-tree/solution/javadi-gui-fang-shi-jie-jue-by-sdwwld/
来源:力扣(LeetCode)

举报

相关推荐

0 条评论