티스토리 뷰
https://www.acmicpc.net/problem/11279
11279번: 최대 힙
첫째 줄에 연산의 개수 N(1 ≤ N ≤ 100,000)이 주어진다. 다음 N개의 줄에는 연산에 대한 정보를 나타내는 정수 x가 주어진다. 만약 x가 자연수라면 배열에 x라는 값을 넣는(추가하는) 연산이고, x가
www.acmicpc.net
👩💻문제 이해
힙 자료구조를 이용해 최대 힙을 구현한다.
파이썬 heapq 모듈은 최소 힙으로 구현되어 있기 때문에 heappush를 할 때 그냥 num이 아닌 (-num, num)으로 구성된 튜플을 이용하게 되면, 튜플의 첫 번째 원소인 -num이 우선순위로 힙이 구성되므로 최대 힙을 구현할 수 있다.
👩💻heapq 모듈을 사용해 최대 힙 구현 : 성공🌈
from heapq import heappop, heappush
import sys
n = int(input())
heap = []
for i in range(n):
num = int(sys.stdin.readline())
if num != 0:
heappush(heap,(-num, num))
else:
try:
print(heappop(heap)[1])
except:
print(0)'🦖 Programming > Python' 카테고리의 다른 글
| [Python] 백준 알고리즘 2075번 : N번째 큰 수 (1) | 2022.10.07 |
|---|---|
| [Python] 백준 알고리즘 11286번 : 절댓값 힙 (1) | 2022.10.06 |
| [Python] 백준 알고리즘 1927번 : 최소 힙 (0) | 2022.10.05 |
| [Python] 백준 알고리즘 1966번 : 프린터 큐 (0) | 2022.10.04 |
| [Python] 백준 알고리즘 10866번 : 덱 (0) | 2022.10.04 |