백준 1927번: 최소 힙 (Python)

접근

이전 문제인 최대 힙과 동일한 문제였다. 이전 글에 힙에 대한 설명 참고하면 되겠다.

2021.04.05 - [코딩/백준 (Python)] - 백준 11279번: 최대 힙 (Python)

백준 11279번: 최대 힙 (Python)

코드

import sys


def heapappend(heap, num):
    heap.append(num)

    i = len(heap) - 1
    while i > 1:
        if heap[i] < heap[i // 2]:
            tmp = heap[i]
            heap[i] = heap[i // 2]
            heap[i // 2] = tmp
            i = i // 2
        else:
            break


def heappop(heap):
    popvalue = heap[1]
    tmp = heap.pop()

    parent = 1
    child = 2
    while child <= len(heap) - 1:
        if child < len(heap) - 1 and heap[child] > heap[child + 1]:
            child += 1
        if heap[child] >= tmp:
            break
        heap[parent] = heap[child]
        parent = child
        child *= 2

    if len(heap) != 1:
        heap[parent] = tmp
    return popvalue


n = int(sys.stdin.readline())
a = [0]
for _ in range(n):
    m = int(sys.stdin.readline())
    if m == 0:
        if len(a) == 1:
            print(0)
        else:
            print(heappop(a))
    else:
        heapappend(a, m)

더 생각해 볼 것?

...

코드나 내용 관련 조언, 부족한 점 및 질문 언제든 환영합니다!