[코테] 리트코드 트리 108. Convert Sorted Array to Binary Search Tree

108. Convert Sorted Array to Binary Search Tree

문제 링크

https://leetcode.com/problems/convert-sorted-array-to-binary-search-tree/description/

문제 설명

Given an integer array nums where the elements are sorted in ascending order, convert it to a height-balanced binary search tree.

제한 사항

• 1 <= nums.length <= 10^4
• -10^4 <= nums[i] <= 10^4
• nums is sorted in a strictly increasing order.

입출력 예 #1

1. Input : nums = [-10,-3,0,5,9]
2. Output : [0,-3,9,-10,null,5]
3. Explanation : [0,-10,5,null,-3,null,9] is also accepted:

입출력 예 #2

1. Input : nums = [1,3]
2. Output : [3,1]
3. Explanation : [1,null,3] and [3,1] are both height-balanced BSTs.

문제 풀이

정렬된 리스트를 반으로 나누며 작은건 왼쪽 노드, 큰 건 오른쪽 노드로 나누며 트리를 구성한다.

class Solution:
    def sortedArrayToBST(self, nums: List[int]) -> Optional[TreeNode]:
        if not nums:
            return None

        mid = len(nums) // 2

        # 분할 정복으로 이진 검색 결과 트리 구성
        node = TreeNode(nums[mid])
        node.left = self.sortedArrayToBST(nums[:mid])
        node.right = self.sortedArrayToBST(nums[mid+1:])

        return node