[백준 11279] 최대 힙

www.acmicpc.net/problem/11279

11279번: 최대 힙

---

접근

python의 heapq를 사용해보았다.

최대힙이 되도록 heap 자료구조에 넣고 0을 입력할 때마다 최댓값을 출력한다.

 

구현

N개의 수를 입력한다.

0이 입력되었을 때 heap이 비어있으면 0을, 아니라면 최댓값을 출력한다.

이외의 수가 입력된다면 heap 자료구조에 최댓값이 위에 있도록 마이너스 값과 튜플형태로 넣어준다.

import sys
import heapq
input = sys.stdin.readline

N = int(input())
hq = []
for _ in range(N):
    number = int(input())
    if number == 0:
        if not hq:
            print(0)
        else:
            result = heapq.heappop(hq)
            print(result[1])
    else:
        heapq.heappush(hq, (-number, number))