108. 将有序数组转换为二叉搜索树

最新推荐文章于 2023-02-01 10:23:38 发布

北木.

最新推荐文章于 2023-02-01 10:23:38 发布

阅读量122

点赞数

分类专栏：多味的LeetCode 文章标签： leetcode 二叉树

本文链接：https://blog.csdn.net/weixin_43283397/article/details/115057812

版权

多味的LeetCode 专栏收录该内容

211 篇文章 3 订阅

订阅专栏

题目描述：

给你一个整数数组 nums ，其中元素已经按升序排列，请你将其转换为一棵高度平衡二叉搜索树。

高度平衡二叉树是一棵满足「每个节点的左右两个子树的高度差的绝对值不超过 1 」的二叉树。

示例 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,3] 和 [3,1] 都是高度平衡二叉搜索树。

提示：

1 <= nums.length <= 104
$10^4 <= nums[i] <= 10^4$
nums 按严格递增顺序排列

解题思路:

中序遍历，迭代生成左右子树

代码：

# Definition for a binary tree node.
# class TreeNode(object):
#     def __init__(self, val=0, left=None, right=None):
#         self.val = val
#         self.left = left
#         self.right = right
class Solution(object):
    def sortedArrayToBST(self, nums):
        def helper(left, right):
            if left > right:
                return None
            mid = (left + right) // 2
            root = TreeNode(nums[mid])
            root.left = helper(left, mid-1)
            root.right = helper(mid+1, right)
            return root
        return helper(0, len(nums)-1)