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

题目:    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;
    }
}

 

你可能感兴趣的:(算法)