[Python] BOJ 11279번. 최대 힙

11279번. 최대 힙

문제 링크

풀이 코드

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
# 11279번. 최대 힙


import heapq
import sys
input = sys.stdin.readline

n = int(input())
h = []
for i in range(n):
    ii = int(input())
    if ii != 0:
        # heapq는 최소 힙.
        # 최대 힙으로 바꾸기 위해 input을 음수로 바꿔줌
        heapq.heappush(h, (-ii))
    else:
        if not h:
            print(0)
        else:
            print(-1 * heapq.heappop(h))

비고