プログラマ

質問する

にゅうしゅつりょく

コミットコード

import heapq

def solution(operations):
    answer = []
    q = []

    for operation in operations:
        o, num = operation.split()
       
		# I일 경우 num 삽입
        if o == "I":
        	# num을 int로 바꾸지 않으면 음수에 대한 처리가 안됨.
            heapq.heappush(q, int(num))
            continue
        if q:
        	# 최소 힙으로 heappop 하면 됨.
            if num == "-1":
                heapq.heappop(q)
            # q를 정렬하고 마지막 값을 제거했는데, 사실 이렇게 안하고 q.remove(max(q))이렇게 해줘도 됨.
            else:
                q.sort()
                q = q[:-1]
                # 해당 배열을 heapify해줌.
                heapq.heapify(q)

    if q:
        m_in = heapq.heappop(q)
        if q:
            m_ax = max(q)
            answer = [m_ax, m_in]
        # q에 값이 하나밖에 없던 상황, 최댓값과 최솟값이 같은 상황으로 다음과 같이 처리한다.
        else:
            answer = [m_in, m_in]
    else:
        answer = [0, 0]
    return answer