[백준 11279] 최대 힙

Junyoung Park·2022년 2월 26일
0

코딩테스트

목록 보기
99/631
post-thumbnail

1. 문제 설명

최대 힙

2. 문제 분석

파이썬의 heapq 모듈을 통해 최대 힙을 구현하려면 -1을 곱해 음수 min-heap를 사용하자.

  • input()이 아니라 sys.stdin.readline()을 써야 시간 초과가 나지 않는다.

3. 나의 풀이

import heapq
import sys
n = int(input())
heap = []

for _ in range(n):
    num = int(sys.stdin.readline())
    if num == 0:
        if not heap: print(0)
        else:
            print(-1 * heapq.heappop(heap))
    else:
        heapq.heappush(heap, -1 * num)

profile
JUST DO IT

0개의 댓글