和下面两题思路差不多,推荐一起做
深度优先搜索:力扣108. 将有序数组转换为二叉搜索树
深度优先搜索、快慢指针:力扣109. 有序链表转换二叉搜索树
其实就是先中序遍历,把原二叉搜索树转化成有序数组,然后再把有序数组转化为平衡的二叉搜索树
# Definition for a binary tree node.
# class TreeNode:
# def __init__(self, x):
# self.val = x
# self.left = None
# self.right = None
class Solution:
def balanceBST(self, root: TreeNode) -> TreeNode:
#可以中序遍历先转化为有序数组,然后再转化为平衡的二叉树
nums = []
def inorder(root):
if not root:
return
inorder(root.left)
nums.append(root.val)
inorder(root.right)
inorder(root)
def sortedArrayToBST(nums):
if not nums:
return
mid_index = len(nums) // 2
mid = nums[mid_index]
root = TreeNode(mid)
root.left = sortedArrayToBST(nums[:mid_index])
root.right = sortedArrayToBST(nums[mid_index+1:])
return root
return sortedArrayToBST(nums)
时间复杂度:O(N)
空间复杂度:O(N),存放了一个数组