LeetCode 初级 - 将有序数组转换为二叉搜索树

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

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

本题中,一个高度平衡二叉树是指一个二叉树每个节点 的左右两个子树的高度差的绝对值不超过 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 {
        public TreeNode sortedArrayToBST(int[] nums) {
            if(nums.length == 0){
                return null;
            }
            return buildBST(nums,0,nums.length-1);
        }
        private TreeNode buildBST(int[] nums,int start, int end){
            if(start<=end){
                int mid = (start + end) / 2;
                TreeNode root = new TreeNode(nums[mid]);
                root.left = buildBST(nums,start,mid-1);
                root.right = buildBST(nums,mid+1,end);
                return root;
            }else{
                return null;
            }
        }
    }

你可能感兴趣的:(算法,LeetCode,刷题之路)