[백준 파이썬] 11279 최대힙

📌문제 출처

백준 단계별 문제풀이 - 우선순위 큐

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

11279번: 최대 힙

❓ 문제

❗ 풀이

• heap 자료 구조 활용

📗 풀이 코드

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

nums = []
for _ in range(int(input())):
    i = int(input())
    if i :
        heappush(nums,-i)
    else :
        if not nums:
            print(0)
        else :
            print(-heappop(nums))

📗 코드 해설

• heappush로 nums라는 list를 힙구조로 변경하며 원소를 추가한다.
• 이때 음수값 형태로 넣어서 heappop시 절대값이 가장 큰 원소가 나오도록 한다.
• heappop으로 출력한 최대값에 -부호를 붙여 원래의 수를 출력한다.