백준 11286 절대값 힙 - 파이썬

풀이.

최대 힙을 응용하여서 튜플형식에 (abs(n), n)으로 절대값을 삽입해준다.

 

소스코드.

import sys
import heapq

N = int(sys.stdin.readline())

queue = []
for i in range(N):
    n = int(sys.stdin.readline())
    if n != 0:
        heapq.heappush(queue, (abs(n), n))
    elif n == 0:
        try:
            print(heapq.heappop(queue)[1])
        except:
            print(0)