[Python] BOJ 11286번. 절댓값 힙

작성일 | In |

11286번. 절댓값 힙

문제 링크

풀이 코드

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
# 11286번. 절댓값 힙


import heapq
import sys
input = sys.stdin.readline

n = int(input())
h = []
ans = []
for i in range(n):
    ii = int(input())
    if ii != 0:
        # heapq는 최소 힙.
        # 절댓값, 원본을 같이 넣어줌
        heapq.heappush(h, [abs(ii), ii])
    else:
        if not h:
            print(0)
        else:
            print((heapq.heappop(h))[1])

비고