[baekjoon] 백준 1927번(파이썬): 최소 힙

문제

1927번: 최소 힙

알고리즘

- 연산의 개수만큼 반복한다.

- 연산에 대한 정보 x가 0이라면 배열에서 가장 작은 값을 출력한다.

- 배열이 비어 있는데 가장 작은 값을 출력해야 하면 0을 출력한다.

- x가 0이 아니라면 배열에 x를 힙 푸시한다.

코드

import sys
import heapq

n = int(sys.stdin.readline())
heap = [] # 배열

# 연산의 개수만큼 반복한다.
for i in range(n):
    x = int(sys.stdin.readline())

    # x가 0이라면 배열에서 가장 작은 값을 출력한다.
    if x == 0:

        # 배열이 비어 있으면 0을 출력한다.
        if heap:
            print(heapq.heappop(heap))
        else:
            print(0)

    # x가 0이 아니라면 배열에 x를 힙 푸시한다.
    else:
        heapq.heappush(heap, x)

github

GitHub - junjange/CodingTest: 내가 푼 코딩 테스트 문제와 해결법