🦖 Programming/Python

[Python] 백준 알고리즘 11286번 : 절댓값 힙

박낑깡이 2022. 10. 6. 18:20

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

 

11286번: 절댓값 힙

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

www.acmicpc.net

 

 

👩‍💻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,(abs(num), num))
        
    else:
        try:
            print(heappop(heap)[1])
        except:
            print(0)