二叉搜索树:力扣1382. 将二叉搜索树变平衡

1、题目描述:

二叉搜索树:力扣1382. 将二叉搜索树变平衡_第1张图片
二叉搜索树:力扣1382. 将二叉搜索树变平衡_第2张图片

2、题解:

和下面两题思路差不多,推荐一起做
深度优先搜索:力扣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)

3、复杂度分析:

时间复杂度:O(N)
空间复杂度:O(N),存放了一个数组

你可能感兴趣的:(LeetCode高频面试题)