[백준-실버2] 11279번 최대 힙(우선순위큐, 파이썬)

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

11279번: 최대 힙

 

<내 코드>

- 파이썬의 heapq 라이브러리는 디폴트로 최소힙을 제공하기 때문에 최대힙을 사용하기 위해서 힙에 삽입 할때 약간의 트릭을 사용해야 한다. heapq.heappush(heap, (-a,a)) 이렇게 삽입하면 최대힙 구현 가능

- 0은 어차피 가장 큰 수가 되기 때문에 맨 뒤로 가니까 입력으로 0이 들어오면 길이를 따져서 출력하면 된다.

import heapq
import sys
input = sys.stdin.readline

n = int(input())
nums = []
for _ in range(n):
    nums.append(int(input()))
    
h = [] #힙
for i in range(n):
    heapq.heappush(h,(-nums[i], nums[i]))#최대힙
    if nums[i]==0:
        if len(h)==0:
            print(0)
        else:
            print(heapq.heappop(h)[1]) #양수 출력