给你一个整数数组 nums ,其中元素已经按 升序 排列,请你将其转换为一棵 平衡二叉搜索树。
示例 1:
输入:nums = [-10,-3,0,5,9]
输出:[0,-3,9,-10,null,5]
解释:[0,-10,5,null,-3,null,9] 也将被视为正确答案:
示例 2:
输入:nums = [1,3]
输出:[3,1]
解释:[1,null,3] 和 [3,1] 都是高度平衡二叉搜索树。
提示:
1 <= nums.length <= 104
-104 <= nums[i] <= 104
nums 按 严格递增 顺序排列
分治,递归
# Definition for a binary tree node.
# class TreeNode:
# def __init__(self, val=0, left=None, right=None):
# self.val = val
# self.left = left
# self.right = right
class Solution:
def sortedArrayToBST(self, nums: List[int]) -> Optional[TreeNode]:
return self._sortedArrayToBST(nums, 0, len(nums) - 1)
def _sortedArrayToBST(self, nums: List[int], m: int, n: int) -> Optional[TreeNode]:
if m > n:
return None
idx = (m + n) >> 1
node = TreeNode(nums[idx])
node.left = self._sortedArrayToBST(nums, m, idx - 1)
node.right = self._sortedArrayToBST(nums, idx + 1, n)
return node