Leetcode108. 有序数组转为二叉树

DrCrypto 2019-09-01 14:53:07

140

分类专栏： LeetCode

本文链接：https://blog.csdn.net/u011240016/article/details/100183295

版权

Time: 20190901
Type: Easy

题目描述

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

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

示例:

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

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

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/convert-sorted-array-to-binary-search-tree
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

思路

每次找到中点，作为根结点，然后左右划分，递归求解。

代码

# Definition for a binary tree node.
# class TreeNode:
#     def __init__(self, x):
#         self.val = x
#         self.left = None
#         self.right = None

class Solution:
    def sortedArrayToBST(self, nums: List[int]) -> TreeNode:
        def helper(nums, left, right):
            if left >= right:
                return None
            mid = left + (right - left - 1) // 2
            rootVal = nums[mid]
            root = TreeNode(rootVal)
            # 注意这里是用left做左边界，不是0
            root.left = helper(nums, left, mid)
            root.right = helper(nums, mid + 1, right)
            return root
        return helper(nums, 0, len(nums))