Leetcode题解之树（5）将有序数组转换为二叉搜索树

最新推荐文章于 2024-10-15 10:39:14 发布

原创最新推荐文章于 2024-10-15 10:39:14 发布 · 274 阅读

0 ·

CC 4.0 BY-SA版权

算法专栏收录该内容

67 篇文章

订阅专栏

本文介绍如何将升序有序数组转换为高度平衡的二叉搜索树，通过递归法选取数组中间元素作为根节点，确保左右子树高度差不超过1，实现树的平衡。示例代码展示具体实现过程。

题目: https://leetcode-cn.com/explore/interview/card/top-interview-questions-easy/7/trees/51/

题目描述：

将一个按照升序排列的有序数组，转换为一棵高度平衡二叉搜索树。

本题中，一个高度平衡二叉树是指一个二叉树每个节点 的左右两个子树的高度差的绝对值不超过 1。

示例:

给定有序数组: [-10,-3,0,5,9],

一个可能的答案是：[0,-3,9,-10,null,5]，它可以表示下面这个高度平衡二叉搜索树：

      0
     / \
   -3   9
   /   /
 -10  5

思路：递归法。由于是有序的数组。每次取数组中间元素作为新的根节点。

代码：

class Solution {

  public TreeNode sortedArrayToBST(int[] nums) {

        if (nums == null || nums.length == 0) return null;

        return sortedArrayToBST(nums, 0, nums.length-1);
    }

    private TreeNode sortedArrayToBST(int[] nums, int start, int end) {
        if (start > end) return null;
        int mid = (end + start) / 2;
        TreeNode root = new TreeNode(nums[mid]);
        root.left = sortedArrayToBST(nums, start, mid-1);
        root.right = sortedArrayToBST(nums, mid + 1, end);
        return root;
    }
}