[백준-파이썬] 11279: 최대 힙

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

11279번: 최대 힙

예전 코드(시간 초과)

import heapq as hq

N=int(input())
a = []
for _ in range(N):
    n = int(input())
    if n == 0:
        if len(a) == 0:
            print(0)
        else:
            tmp=-hq.heappop(a)  # 루트 노드 값
            print(tmp)
    else:
        hq.heappush(a, -n)

-> 잘 짜서 놀랐다(??)  ㅋㅋ 파이썬의 자료구조 우선순위 큐를 사용하고! heapq는 작은 값이 먼저 나오는 최소 힙인데, 우리는 최대 힙이 필요하니까 -1을 곱한 값을 hq에 heappush하고, -1곱한 값을 출력하고.. 잘했는데?!!

방금 푼 코드(맞았습니다!)

import heapq as hq
import sys

input= sys.stdin.readline

N = int(input())
a = []
for _ in range(N):
    n = int(input())
    if n == 0:
        if len(a) == 0:
            print(0)
        else:
            tmp = -hq.heappop(a)
            print(tmp)
    else:
        hq.heappush(a, -n)

import sys.

input = sys.stdin.readline을 더하니 맞았다..

 

sys 라이브러리에 관한 ..글... 

https://wikidocs.net/33#sys

05-6 라이브러리

https://docs.python.org/ko/3/library/sys.html 

sys — 시스템 특정 파라미터와 함수 — Python 3.10.4 문서