2023.01.28 절댓값 힙 [파이썬]

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

11286번: 절댓값 힙

 

import sys
from heapq import heappop, heappush
input = sys.stdin.readline

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

for _ in range(n):
    val = int(input().strip())
    if val == 0:
        if heap:
            print(heappop(heap)[1])
            continue
        print(0)
        continue
    heappush(heap,(abs(val),val))

heapq 모델을 사용해서 풀었다. 들어가는 값을 튜플로 절대값과 원래 val 값을 넣었으며 abs 가 같을 때 에는 val 이 작은 순서대로 나오게 heappush를 하였다.