[그리디] 프로그래머스 [1차] 셔틀버스 C++

 

https://school.programmers.co.kr/learn/courses/30/lessons/17678

프로그래머스

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

using namespace std;

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

string intToTime(int time) {
    return to_string(time/60/10) + to_string(time/60%10) + ":" + to_string(time%60/10) + to_string(time%60%10);
}

string solution(int n, int t, int m, vector<string> timetable) {
    sort(timetable.begin(), timetable.end());

    int answer = 0;
    int cnt = timetable.size();
    
    int now = 9 * 60; // 셔틀 도착 시간, 09:00부터 n회 t 간격
    int idx = 0; // 크루 대기 순서
    for(int i=0; i<n; i++) {
        int temp = 0; // now에 태울 수 있는 크루원 수

        while (true) {
            if (idx >= cnt) break;
            if (temp >= m) break;
            if (timeToInt(timetable[idx]) <= now) { // if문 안에 idx++
                temp++;
                idx++;
            }
            else break;
        }
    
        if (temp < m) answer = now; // m명 미만일 때, 제일 늦은 도착 시간 = 현재 시간, idx-- 대신 idx-1
        else if (temp >= m) answer = timeToInt(timetable[idx-1]) - 1; // m명 이상일 때, 제일 늦은 도착시간 = 마지막 크루원 도착 시간 - 1

        now += t;
    }
    
    return intToTime(answer);
}