프로그래머스: 이중우선순위큐

문제

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

 

문제 풀이

 주어진 연산 종류에 따라 큐에 연산을 삽입하고 삭제하는 과정을 진행한다. 연산의 종류에 따라 동작하는 과정은 다음과 같다. 명령어에 따라 큐에 숫자를 삽입하고, D 1인 경우 큐에서 최댓값을 삭제한다. 이와 달리 D -1인 경우 큐에서 최솟값을 삭제한다. 이는 별도로 heapq를 사용하지 않더라도 list를 통해 간단히 구현할 수 있다.

 

코드

def solution(operations):
    answer = []

    for cur_op in operations:
        op, num = cur_op.split(' ')

        if op == 'I':
            answer.append(int(num))
        elif answer:
            if num == '1':
                answer.remove(max(answer))
            else:
                answer.remove(min(answer))
                
    return [max(answer), min(answer)] if answer else [0, 0]