[Python/파이썬] 프로그래머스 이중우선순위큐

프로그래머스

코드

import heapq

def solution(operations):
    min_q = []
    max_q = []
    for op in operations:
        if op[0] == "I":
            heapq.heappush(min_q, int(op[2:]))
            heapq.heappush(max_q, -int(op[2:]))
        elif op == "D 1":
            if not min_q:
                continue
            min_q.remove(-heapq.heappop(max_q))
        else:
            if not min_q:
                continue
            max_q.remove(-heapq.heappop(min_q))
    if min_q:
        return [-heapq.heappop(max_q), heapq.heappop(min_q)]
    else:
        return [0,0]

파이썬의 heapq 모듈을 이용하여 풀이하였다. heapq는 최소힙을 구현해주므로 min_q에는 입력받은 숫자를 그대로 넣고, max_q에는 최대값이 가장 앞에 올 수 있도록 입력받은 숫자를 음수로 변환해서 넣어주었다.