11279번 최대 힙

문제링크 : www.acmicpc.net/problem/11279

11279번: 최대 힙

---

---

<알고리즘>

  최대값을 간편하게 pop하기 위해서 입력값에 -1을 곱해서 입력 & 출력시 -1을 곱해서 출력

  위의 요소로 인해 heap의 형태가 실제 값으론 가장 큰 값이 위에 위치하게 됨

 

 

 

메모리 : 34700KB     시간 : 164ms

import sys
import heapq
T = int(sys.stdin.readline())
heap = []
for _ in range(T):
    N = int(sys.stdin.readline())
    if(N==0):
        #힙이 0인지 확인후 최대값 출력 혹은 0출력
        if(len(heap)==0):
            print(0)
        else:
            print(heapq.heappop(heap)*(-1))
    else:
        #힙에 집어넣기
        heapq.heappush(heap,-N)

 

 

+) 참고 힙모듈 사용법 : www.daleseo.com/python-heapq/

[파이썬] heapq 모듈 사용법