将有序数组转换为二叉搜索树Python解法
生活随笔
收集整理的這篇文章主要介紹了
将有序数组转换为二叉搜索树Python解法
小編覺得挺不錯的,現在分享給大家,幫大家做個參考.
給你一個整數數組 nums ,其中元素已經按 升序 排列,請你將其轉換為一棵 高度平衡 二叉搜索樹。
高度平衡 二叉樹是一棵滿足「每個節點的左右兩個子樹的高度差的絕對值不超過 1 」的二叉樹。
來源:力扣(LeetCode)
鏈接:https://leetcode-cn.com/problems/convert-sorted-array-to-binary-search-tree
?
例:
輸入:nums = [-10,-3,0,5,9]
輸出:[0,-3,9,-10,null,5]
解釋:[0,-10,5,null,-3,null,9] 也將被視為正確答案:
解析:
可以使用遞歸,每次去數組中間的值當根節點,左邊的值為左子樹,右邊的值為右子樹,兩邊值得個數不會超過1。
# 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):""":type nums: List[int]:rtype: TreeNode"""def build(left, right): # 構建樹的函數if left > right: # 終止條件return Nonemid = (left + right)/2 # 中間值得下標root = TreeNode(nums[mid]) # 構建根節點root.left = build(left, mid - 1) # 構建左子樹root.right = build(mid + 1, right) # 構建右子樹return root # 返回根節點res = build(0, len(nums) - 1) # 調用函數return res總結
以上是生活随笔為你收集整理的将有序数组转换为二叉搜索树Python解法的全部內容,希望文章能夠幫你解決所遇到的問題。
- 上一篇: PPT 标题文字排版思路分享
- 下一篇: 网信办:严厉打击网上恶意损害企业和企业家