Convert Sorted Array to Binary Search Tree

来源:互联网 发布:软件项目实施计划 编辑:程序博客网 时间:2024/06/06 20:12

Given an array where elements are sorted in ascending order, convert it to a height balanced BST.


这一题因为数组是已经排序好的。所以主要考虑点是在于:

1、作为平衡树,左边的结点的都要比右边的结点小,同时,左边的结点的个数和右边的结点个数之差小于等于1.所以,可以默认为 数组的中位数就是root,小于root的是root的左子树结点,大于root的是root的右子树结点。

接下来就是基本的找中位数的过程。

代码如下:

/** * Definition for a binary tree node. * public class TreeNode { *     int val; *     TreeNode left; *     TreeNode right; *     TreeNode(int x) { val = x; } * } */public class Solution {    public TreeNode sortedArrayToBST(int[] nums) {        if(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 = (start+end)/2;        TreeNode root = new TreeNode(nums[mid]);        root.left = sortedArrayToBST(nums,start,mid-1);        root.right = sortedArrayToBST(nums,mid+1,end);        return root;    }}

注意点如下:

一开始写的时候,初始赋值是nums.length-1,而不是 nums.length;

0 0
原创粉丝点击