[백준] 11279 파이썬(python) : 최대 힙

 

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

11279번: 최대 힙

 

 

import sys, heapq
n = int(sys.stdin.readline())
heap = []
for _ in range(n):
    x = int(sys.stdin.readline())
    if not heap and x == 0: #1
        print(0)
    elif heap and x == 0: #2
        print(heapq.heappop(heap)* -1)
    elif x != 0: #3
        heapq.heappush(heap, -x)

 

#1 : heap이 비어있고 입력값이 0이라면 0을 출력

#2 : 힙이 비어있지 않고 입력값이 0이라면 최대값을 출력

#3 : 입력값이 0이 아니라면 값에 음수를 붙여서 삽입