[swea 2477] 차량 정비소

분류 

시뮬레이션, 구현

요구사항

주어진 안내소와 정비소를 이용한 고객을 찾을 수 있도록 한다.

풀이

문제에서 주어진 대로 구현할 수 있도록 한다.

문제 주어진 조건이 복잡하다

구조체를 활용하여 현재 정비소나 안내소에 들어간 index, time, 안내소 번호, 정비소 번호를 기록할 수 있도록 한다.

1. 현재 시간에 안내소에서 나오는 애들을 빼준다

2. 현재 시간에 안내소에서 들어갈 수 있는 애들을 넣어준다

3. 현재 시간에 정비소에 나오는 애들을 빼준다

4. 현재 시간에 정비소 들어가는 친구들 넣어준다

 

4번에서 고객마다 정비소와 안내소를 기록해 두어서 정답을 출력할 수 있도록 한다

#include <iostream>
#include <queue>
#include <algorithm>
#include <math.h>
#include <string.h>
#include <vector>

using namespace std;


struct go {
	int visit;
	int rindex;
	int time;
	int cindex;
	int index;
};
struct go1 {
	int time;
	int index;
	int reception;
	int repair;
};
int n, m, k, a, b;
int reception_time[10];
int repair_time[10];
int arrive_time[1100];
int anscheck[1100];
go receprion_desk[10];
go repair_desk[10];
int arrive_visit[1100];
vector<go1> ansinfo;
bool com(go1 a, go1 b) {
	if (a.time <b.time) return true;
	else if (a.time == b.time) {
		if (a.reception<b.reception) return true;
	}
	return false;
}
void solve() {
	int ansnum = 0;
	int time = 0;
	deque<go1> v;
	while (true) {
		//	cout<<"time"<<time<<endl;
		if (ansnum == k)
			break;
		//Reception Out
		for (int j = 1; j <= n; j++) {
			if (receprion_desk[j].visit == 1) {
				receprion_desk[j].time++;
				if (reception_time[j] == receprion_desk[j].time) {
					//Repair로 ㄱㄱ
					//		cout<<receprion_desk[j].index <<" recetion out "<<j<<endl;
					v.push_back({ time,receprion_desk[j].index, receprion_desk[j].rindex });
					receprion_desk[j].visit = 0;
					receprion_desk[j].time = 0;
					receprion_desk[j].rindex = 0;
					receprion_desk[j].index = 0;
				}
			}
		}
		//Repair Out
		for (int i = 1; i <= m; i++) {
			if (repair_desk[i].visit == 1) {
				repair_desk[i].time++;
				if (repair_desk[i].time == repair_time[i]) {
					ansinfo.push_back({ 0,repair_desk[i].index,repair_desk[i].rindex,repair_desk[i].cindex });
					//cout<<repair_desk[i].index <<" repair out "<<i<<endl;
					ansnum++;
					repair_desk[i].index = 0;
					repair_desk[i].visit = 0;
					repair_desk[i].time = 0;
					repair_desk[i].rindex = 0;
					repair_desk[i].cindex = 0;
				}
			}
		}
		//Reception In
		for (int i = 1; i <= k; i++)
		{
			//도착 시간이 됫으면 빈 곳을 찾는다.
			if (arrive_visit[i] == 0) {
				if (arrive_time[i] <= time) {
					for (int j = 1; j <= n; j++) {
						if (receprion_desk[j].visit == 0) {
							//	cout<<i <<" recetion in "<<j<<endl;
							receprion_desk[j].visit = 1;
							receprion_desk[j].rindex = j;
							receprion_desk[j].time = 0;
							receprion_desk[j].index = i;
							arrive_visit[i] = 1;
							break;
						}
					}
				}
			}
		}
		sort(v.begin(), v.end(), com);
		//Repair 빈 곳이 있으면 도착자들 넣어준다
		for (int i = 1; i <= m; i++) {
			if (repair_desk[i].visit == 0) {
				if (v.empty())
					break;
				else {
					//	cout<<v[0].index <<" repair In "<<i<<endl;
					repair_desk[i].visit = 1;
					repair_desk[i].rindex = v[0].reception;
					repair_desk[i].index = v[0].index;
					repair_desk[i].time = 0;
					repair_desk[i].cindex = i;
					v.pop_front();
				}
			}
		}
		time++;
		//cout<<endl;
	}
}
int main() {
	int t;
	scanf("%d", &t);
	for (int tc = 1; tc <= t; tc++) {
		int ans = 0;
		memset(reception_time, 0, sizeof(reception_time));
		memset(repair_time, 0, sizeof(repair_time));
		memset(arrive_time, 0, sizeof(arrive_time));
		memset(anscheck, 0, sizeof(anscheck));
		memset(receprion_desk, 0, sizeof(receprion_desk));
		memset(repair_desk, 0, sizeof(repair_desk));
		memset(arrive_visit, 0, sizeof(arrive_visit));
		ansinfo.clear();
		scanf("%d%d%d%d%d", &n, &m, &k, &a, &b);
		for (int i = 1; i <= n; i++) scanf("%d", &reception_time[i]);
		for (int i = 1; i <= m; i++) scanf("%d", &repair_time[i]);
		for (int i = 1; i <= k; i++) scanf("%d", &arrive_time[i]);
		solve();
		for (int i = 0; i <ansinfo.size(); i++) {
			int r = ansinfo[i].reception;
			int c = ansinfo[i].repair;
			if (r == a && c == b) {
				ans += ansinfo[i].index;
			}
		}
		if (ans == 0) printf("#%d %d\n", tc, -1);
		else printf("#%d %d\n", tc, ans);
	}
}