LeetCode——108. 将有序数组转换为二叉搜索树

题目


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

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

示例:

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

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

      0
     / \
   -3   9
   /   /
 -10  5

 解题思路

与二分查找类似,将数组中点作为根节点,左右两部分作为左右子树。 

代码实现 

/**
 * Definition for a binary tree node.
 * public class TreeNode {
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode(int x) { val = x; }
 * }
 */
class Solution {
    
    /**
    * 1、把数组中点作为根节点
    * 2、把左半边数组元素作为左子树,右半边数组元素作为右子树,以此递推。
    */
    public TreeNode sortedArrayToBST(int[] nums) {
        int N = nums.length;
        return sortedArrayToBST(nums, 0, N - 1);
    }
    
    public TreeNode sortedArrayToBST(int[] nums, int lo, int high) {
        // 递归结束条件
        if (lo > high) {
            return null;
        }
        
        int mid = (lo + high) / 2;
        
        TreeNode root = new TreeNode(nums[mid]);
        root.left = sortedArrayToBST(nums, lo, mid - 1);
        root.right = sortedArrayToBST(nums, mid + 1, high);
        return root;
    }
}

猜你喜欢

转载自blog.csdn.net/qq_41722272/article/details/82769747
今日推荐