【LeetCode】108. Convert Sorted Array to Binary Search Tree 解題報告(Python)

題目分析:

將一個按照升序排列的有序數組,轉換爲一棵高度平衡二叉搜索樹。本題中,一個高度平衡二叉樹是指一個二叉樹每個節點 的左右兩個子樹的高度差的絕對值不超過 1。
示例:
給定有序數組: [-10,-3,0,5,9],一個可能的答案是:[0,-3,9,-10,null,5],它可以表示下面這個高度平衡二叉搜索樹:
在這裏插入圖片描述
解題思路:

這一題利用二分查找與遞歸的思想解決,因爲是把一個有序的數組轉換成二叉搜索樹,所以

  1. 數組的中間元素就是二叉搜索樹的根節點
  2. 數組中間元素左側元素是左子樹
  3. 數組中間元素右側元素是右子樹
  4. 遞歸重複1,2,3步驟
  5. 注:這個過程其實也就是中序遍歷逆過程

提交代碼:(遞歸,Runtime: 56 ms, faster than 99.23% )

class Solution:
    def sortedArrayToBST(self, nums: 'List[int]') -> TreeNode:
        if nums == []: return None
        index = len(nums) // 2
        root = TreeNode(nums[index])
        root.left = self.sortedArrayToBST(nums[:index])
        root.right = self.sortedArrayToBST(nums[index + 1:])
        return root
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章