将一个按照升序排列的有序数组,转换为一棵高度平衡二叉搜索树。
本题中,一个高度平衡二叉树是指一个二叉树每个节点 的左右两个子树的高度差的绝对值不超过 1。
示例:
给定有序数组: [-10,-3,0,5,9], 一个可能的答案是:[0,-3,9,-10,null,5],它可以表示下面这个高度平衡二叉搜索树: 0 / \ -3 9 / / -10 5
思路:
每次将中间点作为根节点,将数组左右两边的部分分别递归调用,作为根节点的左右子树。这样可以保证左右子树的高度差的绝对值不超过1.
/**
* Definition for a binary tree node.
* function TreeNode(val) {
* this.val = val;
* this.left = this.right = null;
* }
*/
/**
* @param {number[]} nums
* @return {TreeNode}
*/
var sortedArrayToBST = function(nums) {
if (nums.length === 0) {
return null;
}
if (nums.length === 1) {
return new TreeNode(nums[0])
}
let rootIndex = parseInt(nums.length / 2);
let root = new TreeNode(nums[rootIndex]);
root.left = sortedArrayToBST(nums.slice(0, rootIndex));
root.right = sortedArrayToBST(nums.slice(rootIndex+1))
return root
};