프로그래머스(힙)-이중우선순위큐-Python

문제 설명

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

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

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

제한사항

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

입출력 예

operations	return
[I 16,D 1]	[0,0]
[I 7,I 5,I -5,D -1]	[7,5]

 

---

내 문제풀이

1. 리스트에서 최대, 최소값을 명령어에 따라 계속 찾아 제거해야 하므로, heapq를 사용한다.
2. operations를 반복문을 돌며 하나씩 빼서 확인한다.
3. 명령어의 앞 문자가 I인 경우, heapq(answer)에 값을 추가한다.
4. 이 외 명령어(제거)에 answer이 비어있다면 해당 명령어를 무시한다. 또한 비어있지 않으면 명령어의 뒷 값을 1과 -1로 구분하여 1이면 최대값을 제거, -1이면 최소값을 제거한다.
5. 반복문이 끝난 후 answer에 아직 값이 남아있다면 최대값, 최소값 배열을 반환한다. 
6. answer아 비어있다면 [0,0]을 반환한다.

내 코드 - 파이썬

import heapq
    
def solution(operations):
    answer = []
    
    for op in operations:
        if op.split(" ")[0] == 'I':
            heapq.heappush(answer, int(op.split(" ")[1]))
    
        else:
            if not answer:
                continue
                
            if op.split(" ")[1] == '1':
                answer.pop(-1)
            else:
                answer.pop(0)        
    
    if answer:
        return [max(answer), min(answer)]
    else:
        return [0,0]