[PCCP 모의고사 #2] 3번 - 카페 확장.python

def solution(menu, order, k):
    answer, time_table, t_tm, wait_p = 0, [], 0, 0
    # time_table 작성 [(0, 1), (10, 1), (12, -1)] (시간, 인원 변화)
    for i in range(len(order)):
        time_table.append((k*i, 1))     # i번째 손님 주문 시간
        
        if t_tm > k*i:                  # 주문 시간보다 대기 시간이 더 크면
            t_tm += menu[order[i]]      # 대기 시간에 추가
        else:
            t_tm = k*i + menu[order[i]] # 주문 시간에 추가
        time_table.append((t_tm, -1))
    time_table.sort()
    for tm in time_table:
        wait_p += tm[1]
        if wait_p > answer: answer = wait_p
    return answer