Leetcode108. 有序数组转为二叉树

Time: 20190901
Type: Easy

题目描述

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

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

示例:

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

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

      0
     / \
   -3   9
   /   /
 -10  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))

END.

已标记关键词 清除标记
©️2020 CSDN 皮肤主题: 代码科技 设计师:Amelia_0503 返回首页