[백준][Python] 11286번 절댓값 힙

11286번: 절댓값 힙

 

코드 

import sys
import heapq

n = int(input())
heap = []
for i in range(n):
    x = int(sys.stdin.readline())
    if x == 0:
        if heap:
            print(heapq.heappop(heap)[1])
        else:
            print(0)
    else:
        heapq.heappush(heap,(abs(x),x))

 

설명

heap에 push를 할 때 (절댓값, 원래값)을 같이 넣어줘서 만약 절댓값이 같으면 원래 값이 음수인 것이 먼저 출력되게 해주었다. 

pop을 해줄 때는 heapq.heappop(heap)[1]로 해서 뒤에 있는 원래 값을 출력하게 한다.