Leetcode 108. 将有序数组转换为二叉搜索树 Convert Sorted Array to Binary Search Tree - Python 递归法

原创于 2023-01-25 13:08:16 发布 · 190 阅读

0 ·

本内容遵循CC 4.0 BY-SA版权协议

收录于

Leetcode刷题

二叉树

博客介绍了将有序数组转换为二叉搜索树的算法思路。先找到数组中间元素作为根结点，再用其左右数组构建左右子树。强调递归结束条件，即左索引大于右索引时结束。还指出计算中间元素下标时避免直接相加除2，以防计算错误。

思路：

找到数组中间元素nums[mid]，并将其作为树的根结点root；

依次用nums[mid]左边和右边数组，构建root的左子树、右子树；

注意递归结束条件非常重要：if left>right: return None

当数组左索引>右索引，则意味着数组中的元素都以放入二叉树中

注意计算数组中间元素下标，mid = left + (right-left) // 2

若直接 mid = (left + right) / 2 ，则有可能因left和right的值都接近infinite而发生计算错误。

# 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.traversal(nums, 0, len(nums)-1)
    
    def traversal(self, nums: List[int], left: int, right: int) -> TreeNode:
        if left > right:
            return None
        
        mid = left + (right-left) // 2
        root = TreeNode(nums[mid])
        root.left = self.traversal(nums,left, mid-1)
        root.right = self.traversal(nums, mid+1, right)

        return root

引用：

https://github.com/youngyangyang04/leetcode-master/blob/master/problems/0108.%E5%B0%86%E6%9C%89%E5%BA%8F%E6%95%B0%E7%BB%84%E8%BD%AC%E6%8D%A2%E4%B8%BA%E4%BA%8C%E5%8F%89%E6%90%9C%E7%B4%A2%E6%A0%91.md

标签

#leetcode #算法 #职场和发展