🟩 [백준] [Python] [Class3] 11286번_절댓값 힙

문제

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

11286번: 절댓값 힙

풀이

# 백준 11286번 문제 - 절댓값 힙
import sys
import heapq
heap = []
n = int(sys.stdin.readline())
for i in range(n):
    num = int(sys.stdin.readline())
    if num != 0:
        heapq.heappush(heap, (abs(num), num))
    else:
        if not heap:
            print(0)
        else:
            print(heapq.heappop(heap)[1])

heapq 모듈을 사용하여 문제를 해결하였다. 

heap에 튜플 자료형으로 push를 해주고 튜플의 첫번째 원소를 먼저 비교하기 때문에 절댓값을 씌워주었다. 

 

힙 자료구조는 우선순위 큐에서도 사용되고 매우 중요하니 제대로 알아두도록 하자.