[백준 11279번 ] 최대 힙 (python, heapq)

728x90
반응형

www.acmicpc.net/problem/11279

 

11279번: 최대 힙

첫째 줄에 연산의 개수 N(1≤N≤100,000)이 주어진다. 다음 N개의 줄에는 연산에 대한 정보를 나타내는 정수 x가 주어진다. 만약 x가 자연수라면 배열에 x라는 값을 넣는(추가하는) 연산이고, x가 0이

www.acmicpc.net

heapq 모듈을 쓰면 heappop할 때 기본 작은 순서대로 반환하는데 -를 붙이면 큰 값이 가장 작아지므로 그렇게 넣은 뒤

반환한 값을 다시 -를 붙여서 되돌린다

import sys
from heapq import heappop,heappush
input=sys.stdin.readline

n=int(input())
q=[]
for _ in range(n):
    x=int(input())
    if x==0:
        if not q:
            print(0)
        else:
            print(-heappop(q))
    else:
        heappush(q,-x)
728x90
반응형
TAGS.

Comments