[백준][Python] 11286번 절댓값 힙 - 코팩

https://www.acmicpc.net/problem/11286

11286번: 절댓값 힙

풀이

힙 문제를 풀기 위해서 heapq 라이브러리를 import해서 사용해주었습니다.

힙에는 [절댓값(숫자), 숫자] 형태로 넣어주었습니다. 출력시에는 '숫자'만 출력할 수 있게 인덱스 1을 써서 뽑아줍니다.

# 11286번 절댓값 힙
import sys, heapq
heap = []
k = int(input())
for _ in range(k):
    num = int(sys.stdin.readline())
    if num == 0:
        if heap:
            print(heapq.heappop(heap)[1])
        else:
            print(0)
    else:
        heapq.heappush(heap, [abs(num), num])

출력결과