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

코딩테스트 연습 - 이중우선순위큐

import sys
import heapq

input = sys.stdin.readline

operations = [input().rstrip() for _ in range(int(input()))]


def solution(operations):
    q = []
    for s in operations:
        oper, num = s.split()[0], int(s.split()[1])
        if oper == "I":
            heapq.heappush(q, num)
        elif q and oper == "D":
            if num == 1:
                q.pop()
            else:
                heapq.heappop(q)
    if len(q) == 0:
        return [0, 0]
    return [max(q), min(q)]


print(solution(operations))

 

해설

힙을 이용한다. 최솟값을 빼야하면 heappop, 최댓값을 빼야하면 pop을 시켜준다 모든 명령어 수행이 끝나면 리스트 길이에 따라 [0,0] 또는 [max(q), min(q)]를 반환한다.