알고리즘/백준

[백준] 11286번 절댓값 힙 (파이썬)

알감자 2022. 1. 4. 20:05
 

11286번: 절댓값 힙

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

www.acmicpc.net

 

코드

import heapq
import sys

n = int(input())

queue = []
for _ in range(n):
    x = int(sys.stdin.readline())
    if x == 0:
        if not queue:
            print(0)
        else:
            print(heapq.heappop(queue)[1])
    else:
        heapq.heappush(queue, (abs(x),x))