백준 1927번 최소 힙

백준 1927번 최소 힙 문제이다. 

 

문제는 다음과 같다. 

0이 나오면 배열에서 최소값을 출력하고 이외의 숫자는 배열에 넣어주는 문제이다. 

이번 문제는 전에 풀었던 절댓값 큐 문제와 유사하다. 

오히려 조건이 덜 들어간 쉬운 버전인 것 같다. 

 

음수에 대한 처리와 우선순위에 대한 부분을 없애고 제출을 하니 바로 '맞았습니다' 가 나왔다.

 

코드는 다음과 같다. 

import sys
import heapq

n = int(sys.stdin.readline().strip())
arr = []
for i in range(n):

    ins = sys.stdin.readline().strip()
    ins = int(ins)
    if ins == 0:
        if len(arr) == 0:
            print(0)
        else:
            print(arr[0])
            heapq.heappop(arr)
    else:
        heapq.heappush(arr, ins)

성공!