[코테 - py] 최소 힙

https://www.acmicpc.net/problem/1927

 

오늘은 백준 문제이다. 최소 힙을 이용한 문제! heapq를 사용했다. 처음에 계속 시간초과가 나길래 이유가 무엇인지 고민했는데 import sys를 써서 입력을 받으니 바로 통과가 되었다. 요즘 그냥 input을 쓰는게 습관이 되어서 import sys를 사용하지 않았는데 경우에 따라 꼭 사용해야겠다. 그리고 우선순위 큐 문제를 많이 풀어봐야지!

 

import heapq
import sys

num = int(sys.stdin.readline())

data = []

for i in range(num):
    n = int(sys.stdin.readline())
    if n != 0:
        heapq.heappush(data, n)
    else:
        if (len(data)) > 0:
            print(heapq.heappop(data))
        else:
            print(0)