[C++] 삼성 2477. [모의 SW 역량테스트] 차량 정비소

접수 창구와 정비 창구의 작동을 구현한다.

접수 창구가 비어있다면 차례대로 고객을 넣어준다.

접수 창구에서 일이 끝난 고객은 정비 창구로 바로 이동한다. 정비 창구가 꽉 차있으면 기다린다.

가장 늦게 도착한 고객의 시간보다 지난 시간을 클 경우, 접수 창구와 정비 창구가 모두 비어있다면 작동을 멈춘다.

#include <iostream>
#include <queue>
#include <cstring>
using namespace std;
 
struct INF {
    int client_num, time;
};

struct Client {
    int rec, rep, arr_time;
} client[1001];
 
INF reception[10], repair[10];
int rec_time[10], rep_time[10];
int N, M, K, A, B;
 
void process_work() {
    memset(reception, 0, sizeof(reception));
    memset(repair, 0, sizeof(repair));
    queue<int> rep_wait;
    int time = 0;
    int index = 1;
    while (1) {
        for (int i = 1; i <= N; i++) {
            if (reception[i].time > 0) {
                reception[i].time--;
                if (reception[i].time == 0) {
                    rep_wait.push(reception[i].client_num);
                    reception[i].client_num = 0;
                }
            }
        }
        while (index <= K and client[index].arr_time <= time) {
            bool flag = false;
            for (int i = 1; i <= N; i++) {
                if (reception[i].time == 0) {
                    reception[i].time = rec_time[i];
                    reception[i].client_num = index;
                    client[index++].rec = i;
                    flag = true;
                    break;
                }
            }
            if (!flag) break;
        }
        for (int i = 1; i <= M; i++) {
            if (repair[i].time > 0) {
                repair[i].time--;
            }
        }
        while (!rep_wait.empty()) {
            bool flag = false;
            for (int i = 1; i <= M; i++) {
                if (repair[i].time == 0) {
                    int now = rep_wait.front(); rep_wait.pop();
                    repair[i].time = rep_time[i];
                    client[now].rep = i;
                    flag = true;
                    break;
                }
            }
            if (!flag) break;
        }
        time++;
        if (time > client[K].arr_time) {
            int cnt = 0;
            for (int i = 1; i <= N; i++) {
                if (reception[i].time == 0) cnt++;
            }
            for (int i = 1; i <= M; i++) {
                if (repair[i].time == 0) cnt++;
            }
            if (cnt == N + M) break;
        }
    }
}
int get_client_num() {
    int ret = -1;
    for (int i = 1; i <= K; i++) {
        if (client[i].rec == A and client[i].rep == B) {
            if (ret == -1) ret = 0;
            ret += i;
        }
    }
    return ret;
}
 
int main() {
    int T;
    cin >> T;
    for (int t = 1; t <= T; t++) {
       cin >> N >> M >> K >> A >> B;
 for (int i = 1; i <= N; i++) cin >> rec_time[i];
 for (int i = 1; i <= M; i++) cin >> rep_time[i];
 for (int i = 1; i <= K; i++) cin >> client[i].arr_time;
       process_work();
        cout << '#' << t << ' ' << get_client_num() << endl;
    }
    return 0;
}