코딩 테스트 - 과제 진행하기

과제 진행하기

문제 링크 : 과제 진행하기

문제 설명

과제를 받은 루는 다음과 같은 순서대로 과제를 하려고 계획을 세웠습니다.

• 과제는 시작하기로 한 시각이 되면 시작합니다.
• 새로운 과제를 시작할 시각이 되었을 때, 기존에 진행 중이던 과제가 있다면 진행 중이던 과제를 멈추고 새로운 과제를 시작합니다.
• 진행중이던 과제를 끝냈을 때, 잠시 멈춘 과제가 있다면, 멈춰둔 과제를 이어서 진행합니다.
  • 만약, 과제를 끝낸 시각에 새로 시작해야 되는 과제와 잠시 멈춰둔 과제가 모두 있다면, 새로 시작해야 하는 과제부터 진행합니다.
• 멈춰둔 과제가 여러 개일 경우, 가장 최근에 멈춘 과제부터 시작합니다.

과제 계획을 담은 이차원 문자열 배열 plans가 매개변수로 주어질 때, 과제를 끝낸 순서대로 이름을 배열에 담아 return 하는 solution 함수를 완성해주세요.

제한 사항

• 3 ≤ plans의 길이 ≤ 1,000
  • plans의 원소는 [name, start, playtime]의 구조로 이루어져 있습니다.
  • name : 과제의 이름을 의미합니다.
    • 2 ≤ name의 길이 ≤ 10
    • name은 알파벳 소문자로만 이루어져 있습니다.
    • name이 중복되는 원소는 없습니다.
  • start : 과제의 시작 시각을 나타냅니다.
    • "hh:mm"의 형태로 "00:00" ~ "23:59" 사이의 시간값만 들어가 있습니다.
    • 모든 과제의 시작 시각은 달라서 겹칠 일이 없습니다.
    • 과제는 "00:00" ... "23:59" 순으로 시작하면 됩니다. 즉, 시와 분의 값이 작을수록 더 빨리 시작한 과제입니다.
  • playtime : 과제를 마치는데 걸리는 시간을 의미하며, 단위는 분입니다.
    • 1 ≤ playtime ≤ 100
    • playtime은 0으로 시작하지 않습니다.
  • 배열은 시간순으로 정렬되어 있지 않을 수 있습니다.
• 진행중이던 과제가 끝나는 시각과 새로운 과제를 시작해야하는 시각이 같은 경우 진행중이던 과제는 끝난 것으로 판단합니다.

입출력 예

plans	result
[["korean", "11:40", "30"], ["english", "12:10", "20"], ["math", "12:30", "40"]]	["korean", "english", "math"]
[["science", "12:40", "50"], ["music", "12:20", "40"], ["history", "14:00", "30"], ["computer", "12:30", "100"]]	["science", "history", "computer", "music"]
[["aaa", "12:00", "20"], ["bbb", "12:10", "30"], ["ccc", "12:40", "10"]]	["bbb", "ccc", "aaa"]

풀이 방법

• 멈춰둔 과제를 진행할 때, 가장 최근에 멈춰둔 과제를 먼저 진행한다고 하니, 스택 자료 구조를 활용해서 멈춰둔 과제를 저장해야 할 것으로 보인다.
• 먼저 시간 순으로 정렬되어 있지 않을 수 있다고 하니, 먼저 시작 시간을 기준으로 정렬을 했다. 그리고 다음 과제 시작하는 시각까지의 시간을 계산한 후에 해당하는 시간에 과제를 끝낼 수 있다면, 해당 과제를 끝낸 후에 스택을 검사하여 처리할 수 있는 과제를 진행할 것이다. 만약 해당하는 시간에 과제를 못 끝낸다면, 진행할 수 있는 시간을 빼고 난 다음에 스택에 넣어서 다음에 여유 시간을 기다려야 할 것이다.
• 마지막으로 마지막 과제는 무조건 진행할 수 있으니 진행한 후에, 스택에 남아있는 과제를 순서대로 해결하면 될 것으로 보인다.
  • 시간복잡도는 정렬하는 데 O(N*logN)의 시간복잡도와, 과제를 진행하는 것을 계산하는 O(N)의 시간복잡도가 걸릴 것으로 보여서 최종적으로 O(N*logN)의 시간복잡도가 걸려서 N이 최대 1,000이기 때문에 알맞은 알고리즘으로 보인다.

구현

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

using namespace std;

struct assignment{
    string name;
    int remaining;
};

int timeToInt(string s){
    return stoi(s.substr(0,2)) * 60 + stoi(s.substr(3, 2));
}

vector<string> solution(vector<vector<string>> plans) {
    vector<string> answer;
    stack<assignment> assignStack;
    
    // 과제 시작 시각을 기준으로 정렬
    sort(plans.begin(), plans.end(), [](vector<string> a, vector<string> b){
        return a[1] < b[1];
    });
    
    // 다음 과제 시작 시각까지 과제 진행
    for(int i = 0; i < plans.size() - 1; i++){
        int spendTime = timeToInt(plans[i + 1][1]) - timeToInt(plans[i][1]);
        
        if (spendTime >= stoi(plans[i][2])){ // 시간 안에 과제를 진행할 수 있는 경우
            spendTime -= stoi(plans[i][2]);
            answer.push_back(plans[i][0]);
            
            while(!assignStack.empty() && spendTime > 0){
                assignment node = assignStack.top(); assignStack.pop();
                if (node.remaining > spendTime){
                    node.remaining -= spendTime;
                    assignStack.push(node);
                    break;
                } else {
                    spendTime -= node.remaining;
                    answer.push_back(node.name);
                }
            }
        } else { // 시간 안에 과제를 진행할 수 없는 경우
            assignStack.push({plans[i][0], stoi(plans[i][2]) - spendTime});
        }
    }
    
    // 남아있는 과제 진행
    answer.push_back(plans[plans.size() - 1][0]);
    
    while(!assignStack.empty()){
        assignment node = assignStack.top(); assignStack.pop();
        answer.push_back(node.name);
    }
    
    return answer;
}