[Programmers]Python_이중우선순위큐

문제

문제 설명

이중 우선순위 큐는 다음 연산을 할 수 있는 자료구조를 말합니다.

명령어	수신 탑(높이)
I 숫자	큐에 주어진 숫자를 삽입합니다.
D 1	큐에서 최댓값을 삭제합니다.
D -1	큐에서 최솟값을 삭제합니다.

이중 우선순위 큐가 할 연산 operations가 매개변수로 주어질 때, 모든 연산을 처리한 후 큐가 비어있으면 [0,0] 비어있지 않으면 [최댓값, 최솟값]을 return 하도록 solution 함수를 구현해주세요.

제한사항

• operations는 길이가 1 이상 1,000,000 이하인 문자열 배열입니다.
• operations의 원소는 큐가 수행할 연산을 나타냅니다.
  • 원소는 “명령어 데이터” 형식으로 주어집니다.- 최댓값/최솟값을 삭제하는 연산에서 최댓값/최솟값이 둘 이상인 경우, 하나만 삭제합니다.
• 빈 큐에 데이터를 삭제하라는 연산이 주어질 경우, 해당 연산은 무시합니다.

코드

import heapq
def solution(operations):
    heap = []
    for operation in operations:
        operator, number = operation.split() #명령어와 숫자로 나누어준다
        number = int(number) #숫자는 string으로 되어있으니 integer로 바꿔주자
        
        #명령어가 I면 숫자를 삽입해주어야한다
        if operator == "I":
            #heapq의 heappush로 heap에 number를 집어넣음으로서 오름차순에 맞게 정렬된다
            heapq.heappush(heap, number)
        elif heap:
            #number가 -1이면 최솟값을 삭제한다
            if number == -1:
                #heapq의 heappop을 사용함으로서 가장 작은 값을 뺀다
                heapq.heappop(heap)
            else:
                #number가 1이면 최댓값을 삭제해야한다
                heap.remove(max(heap))
                
    #heap에 아무것도 없다면 [0, 0]을 return한다                
    if not heap:
        return [0, 0]
        
    return [max(heap), heapq.heappop(heap)] #heap의 최댓값과 최솟값을 return

진행 과정

예제 #1

operations = ["I 7", "I 5", "I -5", "D -1"]

 

1) heap = []

2) heap = [7] #operator = "I"이므로 삽입

3) heap = [5, 7] #operator = "I"이므로 삽입이고 5가 7보다 작은 값이므로 heappush에의해서 왼쪽에 삽입된다

4) heap = [-5, 5, 7] #operator = "I"이므로 삽입이고 -5가장 작은 값이므로 가장 왼쪽에 삽입된다.

5) heap = [5, 7] #operator = "D"이고 number = -1이므로 heappop을 사용하여 최솟값을 빼야한다.

6) return [5, 7]

 

 

어렵지않은 문제이다. 굳이 heapq를 사용하지않고 min()과 max()만 알고있어도 충분히 풀 수 있다.

하지만 시간복잡도를 따지면 heapq를 사용하는 것이 효율적이다.