백준 11286 절댓값 힙 - 파이썬

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

11286번: 절댓값 힙

 

최소 힙으로 해결할 수 있는 문제이다.

 

힙에 [절댓값, 부호]를 넣어두고 heappop해서 사용하면 갓-파이썬은 comparator 같은 거 사용하지 않아도

1. 절댓값이 작은 수

2. 절댓값이 같으면 부호가 음인 수

순서대로 pop해준다.

 

from sys import stdin
import heapq

heap = []

for _ in range(int(input())):
    x = int(stdin.readline())

    if x == 0:
        if heap:
            val, sgn = heapq.heappop(heap)
            print(val * sgn)
        else:
            print(0)
    else:
        heapq.heappush(heap, [abs(x), x//abs(x)])