🔻Algorithm/Programmers
[Programmers] 프로그래머스 이중우선순위큐 Python
_니지
2024. 5. 26. 20:17
https://school.programmers.co.kr/learn/courses/30/lessons/42628
프로그래머스
코드 중심의 개발자 채용. 스택 기반의 포지션 매칭. 프로그래머스의 개발자 맞춤형 프로필을 등록하고, 나와 기술 궁합이 잘 맞는 기업들을 매칭 받으세요.
programmers.co.kr
import heapq
def solution(operations):
min_heap = []
max_heap = []
answer = []
for i in operations:
op, data = i.split(" ")
data = int(data)
if op == "I":
heapq.heappush(min_heap, data)
heapq.heappush(max_heap, -data)
elif len(min_heap) == 0:
continue
elif data == 1:
heapq.heappop(max_heap)
min_heap.pop()
elif data == -1:
heapq.heappop(min_heap)
max_heap.pop()
if len(min_heap) < 1:
answer.append(0)
answer.append(0)
else:
answer.append(-heapq.heappop(max_heap))
answer.append(heapq.heappop(min_heap))
return answer
728x90
반응형