[Codetree] 정수 명령 처리 6 Python

https://www.codetree.ai/missions/8/problems/process-numeric-commands-6/description

코드트리 | 코딩테스트 준비를 위한 알고리즘 정석

 

import sys
from heapq import heappush, heappop


input = sys.stdin.readline

# 최대힙으로
heap = []


n = int(input())

for _ in range(n):
    cmd = input().rstrip().split()

    if cmd[0] == 'push':
        heappush(heap, -int(cmd[1]))

    elif cmd[0] == 'pop':
        top = heappop(heap)
        print(-top)

    elif cmd[0] == 'size':
        print(len(heap))

    elif cmd[0] == 'empty':
        if heap:
            print(0)
        else:
            print(1)

    elif cmd[0] == 'top':
        print(-heap[0])