SWEA 2477 차량 정비소(with Python)

T = int(input())

for tc in range(1, 1+T):
    # N: 접수 창구 개수
    # M: 정비 창구 개수
    # K: 방문 고객 수
    # A, B: 지갑 두고 간 사람이 방문한 창구 번호
    # a: 각 접수 창구 걸리는 시간 list
    # b: 각 정비 창구 걸리는 시간 list
    # t: 각 고객이 정비소를 방문한 시간 list
    N, M, K, A, B = map(int, input().split())
    a = list(map(int, input().split()))
    b = list(map(int, input().split()))
    t = list(map(int, input().split()))
    t_idx = 0
    len_t = len(t)

    # 방문 기록 저장 배열
    visitor_info = [[-1, -1] for _ in range(K)]

    reception_desks = [-1] * N
    repair_desks = [-1] * M

    reception_waiting_line = []
    repair_waiting_line = []

    time = 0

    flag = True
    while flag:
        # 0. 정비 창구에서 볼일 다 본 사람 out
        for i in range(M):
            if repair_desks[i] != -1 and repair_desks[i][1] == 0:
                repair_desks[i] = -1
        # 1. 접수 창구에서 볼일 다 본 사람 '정비 창구 대기열'로 옮기기
        for i in range(N):
            if reception_desks[i] != -1 and reception_desks[i][1] == 0:
                repair_waiting_line.append(reception_desks[i][0])
                reception_desks[i] = -1
        # 2. 이번 시간에 들어온 사람 '접수 창구 대기열'로 옮기기
        for i in range(t_idx, len_t):
            if t[i] == time:
                reception_waiting_line.append(i)
            else:
                t_idx = i
                break
        # 3. 정비 창구 비어있는 곳에 대기열 사람 옮기기
        for i in range(M):
            if repair_desks[i] == -1 and repair_waiting_line:
                visitor_idx = repair_waiting_line.pop(0)
                repair_desks[i] = [visitor_idx, b[i]-1]
                # 몇번 정비 창구 방문했는지 기록하기
                visitor_info[visitor_idx][1] = i
            elif repair_desks[i] != -1:
                repair_desks[i][1] -= 1
        # 4. 접수 창구 비어있는 곳에 대기열 사람 옮기기
        for i in range(N):
            if reception_desks[i] == -1 and reception_waiting_line:
                visitor_idx = reception_waiting_line.pop(0)
                reception_desks[i] = [visitor_idx, a[i]-1]
                # 몇번 접수 창구 방문했는지 기록하기
                visitor_info[visitor_idx][0] = i
            elif reception_desks[i] != -1:
                reception_desks[i][1] -= 1
        # 5. 모든 창구 사람 남은 시간 -1 해주기, 위 3, 4에서 함께 처리 해줌

        time += 1
        if time > K:
            for i in range(K):
                if visitor_info[i][1] == -1:
                    break
            else:
                flag = False

    answer = 0
    for i in range(K):
        if visitor_info[i][0] == A-1 and visitor_info[i][1] == B-1:
            answer += i+1

    if answer == 0:
        answer = -1

    print(f'#{tc} {answer}')