[백준/1927/파이썬(Python3)] 최소힙

문제 출처

1927번: 최소 힙

 

풀이 코드

import heapq
import sys

N = int(input())
heap=[]

for i in range(N):
    x = int(sys.stdin.readline())
    if x!=0:
        heapq.heappush(heap, x)
    else:
        if not heap:
            print(0)
        else:
            print(heapq.heappop(heap))

 

for문에서 입력값을 받을 때 x=int(input()) 을 썼더니 시간초과가 났습니다.

x=int(sys.stdin.readline()) 을 쓰니까 해결 됐습니다.

 

heapq.heappush(heap, x) 말고

heap.append(x) 를 써도 되지 않을까 생각해서 시도해봤지만, append를 쓰면 틀립니다.

 

 

힙 개념 확인하기 ↓

https://data-analysis-expertise.tistory.com/101

[자료구조/파이썬(Python3)] 힙 (Heap)