BOJ/백준-1927-python

문제

풀이

• 최소힙을 구현하는 문제이다.
• 최소힙을 구현하기 위해 heapq 모듈을 불러왔다.
• 각 조건문에 맞춰 해당 연산문을 수행하면 쉽게 구현할 수 있는 문제이다.

코드

# https://www.acmicpc.net/problem/1927
# boj, 1927: 최소 힙, python3
import heapq
import sys

input = sys.stdin.readline

def solve(orders):
    # 최소 힙을 구현하기 위해 heap 초기화
    heap = []

    for x in orders:
        # x가 자연수라면 배열에 x라는 값을 넣는 연산을 수행
        if x != 0:
            heapq.heappush(heap, x)
        # x가 0이라면
        else:
            # 만약 배열이 비어 있는 경우인데 가장 작은 값을 출력하라고 한 경우에는 0을 출력
            if not heap:
                print(0)
            # 배멸에서 가장 작은 값을 출력하고 그 값을 배열에서 제거하는 연산을 수행
            else:
                print(heapq.heappop(heap))


if __name__ == '__main__':
    n = int(input())
    x = [int(input()) for _ in range(n)]

    solve(x)

결과

출처 & 깃허브

BOJ 1927
GITHUB