当前位置：首页 > article >正文

LeetCode108 将有序数组转换为二叉搜索树

article 2024/11/5 6:45:32

题目

给你一个整数数组 nums ，其中元素已经按 升序 排列，
请你将其转换为一棵平衡二叉搜索树。

示例

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

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

解题思路
1. 利用二分查找的思想，每找到一个元素，就将其构造成一个节点。并且因为数组本身已经排序，直接构造即可。

代码(Java)

class Solution {
    public TreeNode sortedArrayToBST(int[] nums) {
        if (nums == null) {
            return null;
        }
        if (nums.length == 1) {
            return new TreeNode(nums[0]);
        }
        return sortedArrayToBST(nums, 0, nums.length - 1, (nums.length - 1) / 2);
    }
    public TreeNode sortedArrayToBST(int[] nums, int left, int right, int mid) {
        if (left > right) {
            return null;
        }
        TreeNode root = new TreeNode(nums[mid]);
        root.left = sortedArrayToBST(nums, left, mid - 1, (left + mid - 1) / 2);
        root.right = sortedArrayToBST(nums, mid + 1, right, (mid + 1 + right) / 2);
        return root;
    }
}

查看全文

http://www.kler.cn/a/273618.html