leetcode之Convert Sorted Array to Binary Search Tree

howtogetout

于 2016-01-12 12:11:13 发布

阅读量252

点赞数

CC 4.0 BY-SA版权

本文链接：https://blog.youkuaiyun.com/howtogetout/article/details/50503177

本文介绍了一种将有序数组转换为平衡二叉搜索树的方法，并提供了详细的Python实现代码。该方法通过选取数组中间元素作为根节点，递归地构建左右子树。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

这题采用的做法是将中间的数作为root，而左子树又是下一个递归的根，依次递归来求解。代码如下：

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

class Solution(object):
    def sortedArrayToBST(self, nums):
        """
        :type nums: List[int]
        :rtype: TreeNode
        """
        length = len(nums)
        if length == 0:
            return None
        if length == 1:
            return TreeNode(nums[0])
        if length == 2:
            root = TreeNode(nums[0])
            root.right = TreeNode(nums[1])
        if length == 3:
            root = TreeNode(nums[1])
            root.left = TreeNode(nums[0])
            root.right = TreeNode(nums[2])
            return root
        root = TreeNode(nums[length / 2])
        root.left = self.sortedArrayToBST(nums[:length / 2])
        root.right = self.sortedArrayToBST(nums[length / 2 + 1:])
        return root