본문 바로가기
백준 알고리즘

파이썬) 백준 알고리즘 | 11279번 : 최대 힙

by 코딩새내기_ 2022. 2. 6.

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

 

11279번: 최대 힙

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

www.acmicpc.net

백준 11279 '최대 힙' 문제입니다.

heapq를 이용하면 되는데 heapq는 최소를 반환하는 모듈이라서 간단한 조작이 필요합니다.

heapq.heappush(heap, (-a,a)) 처럼 튜플형식으로 저장하여 최대값을 반환하도록 할 수 있습니다.

import sys
import heapq
input = sys.stdin.readline
heap = []
N = int(input())
for i in range(N):
    a = int(input())
    if a == 0 :
        if len(heap) == 0:
            print(0)
        else:
            print(heapq.heappop(heap)[1])
    else:
        heapq.heappush(heap, (-a,a))

댓글