Talentd
FleetCode

1382. Balance a Binary Search Tree

Medium86.3% acceptance5 companies

Given the root of a binary search tree, return a balanced binary search tree with the same node values. If there is more than one answer, return any of them.

A binary search tree is balanced if the depth of the two subtrees of every node never differs by more than 1.

Example 1:

Input: root = [1,null,2,null,3,null,4,null,null]
Output: [2,1,3,null,null,null,4]
Explanation: This is not the only correct answer, [3,1,4,null,2] is also correct.

Example 2:

Input: root = [2,1,3]
Output: [2,1,3]

Constraints:

  • The number of nodes in the tree is in the range [1, 104].
  • 1 <= Node.val <= 105

Hints open one at a time

Hint 1
Convert the tree to a sorted array using an in-order traversal.
Hint 2
Construct a new balanced tree from the sorted array recursively.
12:23
Still stuck?Read 3 approaches, or watch a walkthrough.
Open solution
Loading editor…
1# Definition for a binary tree node.
2# class TreeNode:
3# def __init__(self, val=0, left=None, right=None):
4# self.val = val
5# self.left = left
6# self.right = right
7class Solution:
8 def balanceBST(self, root: Optional[TreeNode]) -> Optional[TreeNode]:
9
Starter codePython 3Spaces: 4
Examples
Input
[1,null,2,null,3,null,4]
Input
[2,1,3]
Your code is saved on this device.Solution