99클럽 코테 스터디 7일차 TIL 과제 진행하기

과제 진행하기

문제설명

과제를 받은 루는 다음과 같은 순서대로 과제를 하려고 계획을 세웠습니다.
    1. 과제는 시작하기로 한 시각이 되면 시작합니다.
    2. 새로운 과제를 시작할 시각이 되었을 때, 기존에 진행 중이던 과제가 있다면 진행 중이던 과제를 멈추고 새로운 과제를 시작합니다.
    3. 진행중이던 과제를 끝냈을 때, 잠시 멈춘 과제가 있다면, 멈춰둔 과제를 이어서 진행합니다.
        3-1. 만약, 과제를 끝낸 시각에 새로 시작해야 되는 과제와 잠시 멈춰둔 과제가 모두 있다면, 새로 시작해야 하는 과제부터 진행합니다.
    4. 멈춰둔 과제가 여러 개일 경우, 가장 최근에 멈춘 과제부터 시작합니다.
과제 계획을 담은 이차원 문자열 배열 plans가 매개변수로 주어질 때, 과제를 끝낸 순서대로 이름을 배열에 담아 return 하는 solution 함수를 완성해주세요.

문제풀이

• 가장 최근에 진행했떤 문제를 다시 진행하는 부분에서 stack 자료형을 써야 함을 알 수 있다.
• 시간이 00:00 ~ 23:59 로 표기되기때문에 시간과 분을 합쳤을 경우 가장 작은 것이 먼저 진행된다. 그래서 우선적으로 시간을 기준으로 정렬해줬다
• 과제를 끝난 후에 시간이 남으면 남은 시간동안 미뤄뒀던 과제를 다시 진행할 수 있음을 유의해서 진행해야 한다.

코드

#include <string>
#include <vector>
#include <algorithm>
#include <tuple>
#include <stack>

using namespace std;

vector<string> solution(vector<vector<string>> plans) {
    vector<string> answer;

    vector<tuple<int, int, int, string>> sorted_vector;
    for (int i=0; i<plans.size(); i++) {

        int tmp_h = stoi(plans[i][1].substr(0, 2));
        int tmp_m = stoi(plans[i][1].substr(3, 2));
        sorted_vector.push_back(make_tuple(tmp_h, tmp_m, stoi(plans[i][2]), plans[i][0]));
    }

    sort(sorted_vector.begin(), sorted_vector.end());

    stack<tuple<int, int, int, string>> st;
    for (int i=0; i<sorted_vector.size(); i++) {
        if (st.empty()) st.push(sorted_vector[i]);
        else {
            tuple<int, int, int, string> tmp1, tmp2;
            tmp1 = st.top(); st.pop();
            tmp2 = sorted_vector[i];

            int tm = (get<0>(tmp2)-get<0>(tmp1)) * 60 + (get<1>(tmp2)-get<1>(tmp1)) - get<2>(tmp1);
            if (tm >= 0) {
                answer.push_back(get<3>(tmp1));
                while (tm > 0 && !st.empty()) {
                    if (tm - get<2>(st.top()) >= 0) {
                        tm -= get<2>(st.top());
                        answer.push_back(get<3>(st.top()));
                        st.pop();
                    } else if (!st.empty()) {
                        tuple<int, int, int, string> tmp3 = st.top();
                        st.pop();
                        st.push(make_tuple(get<0>(tmp3), get<1>(tmp3), get<2>(tmp3)-tm, get<3>(tmp3)));
                        break;
                    }
                }
                st.push(tmp2);
            }
            else {
                st.push(make_tuple(get<0>(tmp1), get<1>(tmp1), -tm, get<3>(tmp1)));
                st.push(tmp2);
            }
        }
    }

    while (!st.empty()) {
        answer.push_back(get<3>(st.top()));
        st.pop();
    }

    return answer;
}

회고

• 로직 수행 전에 데이터 처리하는 과정에서 시간과 분을 분리해서 그냥 넣었는데 시간*60 + 분 으로 넣었으면 이후 로직 수행하는 부분에서 더욱수월했을듯
• string 을 다루는 것이 c++에서 익숙하지않아 좀 해맸는데, 이번 과정을 통해 좀 늘 수 있을 것 같다.
• 조건 분기에 좀 더 신경써줘야 할듯