[백준] 11279 최대 힙

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

11279번: 최대 힙

n으로 연산의 횟수를 받아, n 회 입력한 숫자를 리스트 array에 추가

힙 모듈은 min heap만을 지원한다. 따라서 입력값이 0이 아니면 -1을 곱해 힙에 push해야한다. 그러다 보면 최종적으로 절댓값이 가장 작은 수가 힙의 루트에 위치하게 된다.

리스트 array에서 원소가 0 보다 클 경우, 0보다 큰 원소에 -1을 곱해 리스트 ans에 추가

리스트 array에서 원소가 0일 경우일 때마다 결과를 출력하는데

array의 길이가 0이라면 0을 출력

array의 길이가 0보다 크다면 ans에서 heap.heappop을 통해 절댓값이 가장 큰 수를 출력한다.

 

import sys
import heapq

n = int(sys.stdin.readline()) # 연산의 개수

array = []
for i in range(n):
    a = int(sys.stdin.readline())
    array.append(a)


ans = []
for i in array:
    if i > 0:
        heapq.heappush(ans,-i)

    elif i == 0:
        if len(ans) == 0:
            print(0)
        else:
            print( abs(heapq.heappop(ans)))