[LeetCode] 272. Closest Binary Search Tree Value II

class Solution:
    def closestKValues(self, root: Optional[TreeNode], target: float, k: int) -> List[int]:
        h = []
        ret = []

        def dfs(node):
            heapq.heappush(h, (abs(target-node.val), node.val))
            if node.left != None:
                dfs(node.left)
            if node.right != None:
                dfs(node.right)

        dfs(root)
        for i in range(k):
            ret.append(heapq.heappop(h)[1])

        return ret

 270. Closest Binary Search Tree Value 문제의 고난이도 문제다. 이번엔 Target value의 가까운 node를 input으로 주어진 k개만큼 return 해줘야해 Min Heap을 사용했다. Python의 heapq를 사용해 간단하게 구현할 수 있었고, 기본적으로 heapq는 Max Heap이기에 음수를 만들어 heap에 push하는 방식으로 Min Heap을 만들었다.