[소프트웨어 익스퍼트 아카데미][SWEA][모의 SW 역량테스트][ C++][2477번] 차량정비소

swexpertacademy.com/main/code/problem/problemDetail.do?contestProbId=AV6c6bgaIuoDFAXy

SW Expert Academy

 

코드

#include<iostream>
#include <functional>
#include<memory.h>
#include<queue>
using namespace std;
 
struct {
    int Reception;
}Survey[1001];
struct {
    int Rp_p_num;
    int RpT;
} Rp_time[10];
struct {
    int Rc_p_num;;
    int RcT;
} Rc_time[10];
struct pp {
    int ti;
    int P_num;
};
struct aa {
    int Rc_num;
    int numm;
    int a_t;
};
struct cmp {
    bool operator()(aa t, aa u) {
        if (t.a_t == u.a_t) {
            return t.Rc_num > u.Rc_num;
        }
        return t.a_t > u.a_t;
    }
};
 
bool stoop;
bool rpstoop;
 
int N, M, K, A, B; //A지갑을 두고간 접수창고 B정비창고
int Rp_wait[10];  //정비
int Rc_wait[10];  //접수
queue <pp> q;
queue <int> w_q;
priority_queue<aa, vector<aa>, cmp> pq;
int ans;
 
void Repair_func() {
    for (int i = 1; i <= M; i++) {
        if (Rp_time[i].Rp_p_num != 0) {
            if (Rp_time[i].RpT == Rp_wait[i]) { //시간을 다 채웠다면
                if ((i == B) && (Survey[Rp_time[i].Rp_p_num].Reception == A)) { // 정비가 같다면
                    ans += Rp_time[i].Rp_p_num;
                }
                Rp_time[i].RpT = 0;
                Rp_time[i].Rp_p_num = 0;
 
                //for (int i = 1; i <= M; i++) {
                //if ((Rp_time[i].Rp_p_num == 0) && !pq.empty()) {
                //Rp_time[i].Rp_p_num = pq.top().numm;
                //Rp_time[i].RpT = 1;
                //pq.pop();
                //}//비어있다면
                //}
            }
            else
                Rp_time[i].RpT++;
        }
        if (Rp_time[i].Rp_p_num == 0 && !pq.empty()) {
            Rp_time[i].Rp_p_num = pq.top().numm;
            Rp_time[i].RpT = 1;
            pq.pop();
        }//비어있다면
    }//for
}
 
void Reception_func() {
    int timee = 0; stoop = false; rpstoop = false;
    while (1) { //수리고 돌리기 위한 while
        if (stoop == false) {
            if (!q.empty()) {
                while (1) { //나올수 있는거 다나오기
                    if (timee == q.front().ti) {
                        w_q.push(q.front().P_num);
                        q.pop();
                        if (q.empty()) break;
                    }
                    else break;
                }
                for (int i = 1; i <= N; i++) {
                    if (Rc_time[i].Rc_p_num != 0) { //접수창고에 사람이 있다면
                        if (Rc_time[i].RcT == Rc_wait[i]) {  //접수창고에 사람이 있고 대기시간을 모두 채웠다면
                            aa inFo;
                            inFo.a_t = timee;
                            inFo.Rc_num = i;
                            inFo.numm = Rc_time[i].Rc_p_num;
                            pq.push(inFo);
                            Survey[Rc_time[i].Rc_p_num].Reception = i;
                            Rc_time[i].RcT = 0;
                            Rc_time[i].Rc_p_num = 0;
                            //for (int i = 1; i <= N; i++) {
                            //    if ((Rc_time[i].Rc_p_num == 0) && !w_q.empty()) {
                            //        Rc_time[i].Rc_p_num = w_q.front();
                            //        Rc_time[i].RcT = 1;
                            //        w_q.pop();
                            //    } //접수창고에 집어넣기
                            //}//for
                        }//if
                        else
                            Rc_time[i].RcT++;  //접수창고에서 시간을 채우지 못했다면
                    } //if
                    if ((Rc_time[i].Rc_p_num == 0) && !w_q.empty()) {
                        Rc_time[i].Rc_p_num = w_q.front();
                        Rc_time[i].RcT = 1;
                        w_q.pop();
                    } //접수창고에 집어넣기
                }//for
            }//if
            else if (q.empty()) {
                for (int i = 1; i <= N; i++) {
                    if ((Rc_time[i].Rc_p_num == 0) && !w_q.empty()) {
                        Rc_time[i].Rc_p_num = w_q.front();
                        w_q.pop();
                    } //접수창고에 집어넣기
 
                    if (Rc_time[i].Rc_p_num != 0) { //접수창고에 사람이 있다면
                        if (Rc_time[i].RcT == Rc_wait[i]) {  //접수창고에 사람이 있고 대기시간을 모두 채웠다면
                            aa inFo;
                            inFo.a_t = timee;
                            inFo.Rc_num = i;
                            inFo.numm = Rc_time[i].Rc_p_num;
                            pq.push(inFo);
                            Survey[Rc_time[i].Rc_p_num].Reception = i;
                            Rc_time[i].RcT = 0;
                            Rc_time[i].Rc_p_num = 0;
                            //for (int i = 1; i <= N; i++) {
                            //    if ((Rc_time[i].Rc_p_num == 0) && !w_q.empty()) {
                            //        Rc_time[i].Rc_p_num = w_q.front();
                            //        Rc_time[i].RcT = 1;
                            //        w_q.pop();
                            //    } //접수창고에 집어넣기
                            //}//for
                        }//if
                        else
                            Rc_time[i].RcT++;  //접수창고에서 시간을 채우지 못했다면
                    }//if
                    if ((Rc_time[i].Rc_p_num == 0) && !w_q.empty()) {
                        Rc_time[i].Rc_p_num = w_q.front();
                        Rc_time[i].RcT = 1;
                        w_q.pop();
                    } //접수창고에 집어넣기
                }//for
                if (w_q.empty()) {
                    for (int i = 1; i <= N; i++) {
                        if (Rc_time[i].Rc_p_num != 0) {
                            stoop = false;
                            break;
                        }
                        else if (Rc_time[i].Rc_p_num == 0) {
                            stoop = true;
                        }
                    }//for
                }
            } // else if
        }
        Repair_func();//수리소 역활
        if (stoop == true) {
            if (pq.empty()) {
                for (int i = 1; i <= M; i++) {
                    if (Rp_time[i].Rp_p_num != 0) {
                        rpstoop = false;
                        break;
                    }
                    else if (Rp_time[i].Rp_p_num == 0) {
                        rpstoop = true;
                    }
                }//for
            }
        }//while문 탈출 하나 더만들어야 함
        if (rpstoop == true) break;
        //cout << "시간: " << timee << endl << "Rc: " << Rc_time[1].Rc_p_num << "," << Rc_time[2].Rc_p_num << "," << Rc_time[3].Rc_p_num << "," << Rc_time[4].Rc_p_num << endl << "Rp: " << Rp_time[1].Rp_p_num << "," << Rp_time[2].Rp_p_num << endl << endl;
        timee++;
 
    } // while (1)
      //cout << "aaaaaaaaaaaaaaaa";
}//접수 창고 처리 함수
 
int main() {
    ios_base::sync_with_stdio(false);
    cin.tie(0); cout.tie(0);
    int T;
    cin >> T;
    for (int t = 1; t <= T; t++) {
        memset(Rc_time, 0, sizeof(Rc_time)); memset(Rc_wait, 0, sizeof(Rc_wait));
        memset(Rp_time, 0, sizeof(Rp_time)); memset(Rp_wait, 0, sizeof(Rp_wait));
        while (!q.empty()) q.pop();
        while (!w_q.empty()) w_q.pop();
        while (!pq.empty()) pq.pop();
        ans = 0;
        N = 0; M = 0; K = 0; A = 0; B = 0;
 
        cin >> N >> M >> K >> A >> B;
        for (int i = 1; i <= N; i++) {
            cin >> Rc_wait[i];
        }//for N
 
        for (int i = 1; i <= M; i++) {
            cin >> Rp_wait[i];
        }//for M
        for (int i = 1; i <= K; i++) {
            int ttt;
            pp tt;
            tt.P_num = i;
            cin >> ttt;
            tt.ti = ttt;
            q.push(tt);
        }
        Reception_func();
        if (ans != 0)
            cout << "#" << t << " " << ans << endl;
        if (ans == 0) {
            cout << "#" << t << " -1" << endl;
        }
 
    }//for T
}

해설 

큐를 사용하여 풀었습니다.

문제에서 요구하는 그대로 구현을 해주면 되는 문제였습니다.