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

    技术2023-11-05  105

    1、题目描述:

    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),存放了一个数组

    Processed: 0.009, SQL: 9