SWEA 2930. 힙

https://swexpertacademy.com/main/code/problem/problemDetail.do?contestProbId=AV-Tj7ya3jYDFAXr

SW Expert Academy

 

전체 코드

import heapq

T = int(input())

print_num_list = []
for testcase in range(1, T + 1):
    heap = []
    print_nums = [testcase]
    N = int(input())
    for _ in range(N):
        read_line = input()
        # x를 힙에 넣어라
        if len(read_line) > 2:
            cmd, num = map(int, read_line.split())
            heapq.heappush(heap, -num)
        # x를 힙에서 빼라
        else:
            cmd = int(read_line)
            if len(heap) > 0:
                print_nums.append(-heapq.heappop(heap))
            else:
                print_nums.append(-1)

    print_num_list.append(print_nums)

for print_nums in print_num_list:
    s = list(map(str, print_nums))
    print('#' + ' '.join(s))

 

풀이

출력 숫자를 전부 모아서 한번에 print 해야 타임아웃이 발생하지 않는다.