백준 11286번 "절대값 힙"

문제

백준 번

풀이

절대값이 가장 작은 수를 출력해야하는 최소 힙.
힙에 x의 절대값, x쌍으로 이루어진 튜플을 넣어줌으로 써 절대값을 바탕으로 정렬을 할 수 있도록 한다.

Python 코드

import sys
import heapq
input = sys.stdin.readline

n = int(input())
heap = []

for _ in range(n):
    x = int(input())
    if x != 0:
        heapq.heappush(heap, (abs(x), x))
    else:
        try:
            print(heapq.heappop(heap)[1])
        except:
            print(0)