[힙] 백준 11286번 '절댓값 힙' (Python)

<문제 링크>

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

2887번: 행성 터널

 

---

<문제 풀이>

1. 최솟값이라는 우선순위대로 출력하므로 힙 사용

 

<코드>

#내코드
#17분/40분

import heapq
import sys

input = sys.stdin.readline
result = []
queue = []

n = int(input())
for i in range(n):
    x = int(input())
    if x != 0:
        heapq.heappush(queue,[abs(x),x//abs(x)])
    else:
        if queue==[]:
            print(0)
        else:
            result = heapq.heappop(queue)
            print(result[0]*result[1])

#패스트캠퍼스 답
import sys
input = sys.stdin.readline
import heapq

heap = []
n = int(input())
for i in range(n):
	x = int(input())
    if x==0:
    	if len(heap)==0:
        	print(0)
        else:
        	absolute, original = heapq.heappop(heap)
            print(original)
    else:
    	heapq.heappush(heap,(abs(x),x))