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

문제

문제 링크

카카오에서는 무료 셔틀버스를 운행하기 때문에 판교역에서 편하게 사무실로 올 수 있다. 카카오의 직원은 서로를 '크루'라고 부르는데, 아침마다 많은 크루들이 이 셔틀을 이용하여 출근한다.

이 문제에서는 편의를 위해 셔틀은 다음과 같은 규칙으로 운행한다고 가정하자.

• 셔틀은 09:00부터 총 n회 t분 간격으로 역에 도착하며, 하나의 셔틀에는 최대 m명의 승객이 탈 수 있다.
• 셔틀은 도착했을 때 도착한 순간에 대기열에 선 크루까지 포함해서 대기 순서대로 태우고 바로 출발한다. 예를 들어 09:00에 도착한 셔틀은 자리가 있다면 09:00에 줄을 선 크루도 탈 수 있다.

일찍 나와서 셔틀을 기다리는 것이 귀찮았던 콘은, 일주일간의 집요한 관찰 끝에 어떤 크루가 몇 시에 셔틀 대기열에 도착하는지 알아냈다. 콘이 셔틀을 타고 사무실로 갈 수 있는 도착 시각 중 제일 늦은 시각을 구하여라.

단, 콘은 게으르기 때문에 같은 시각에 도착한 크루 중 대기열에서 제일 뒤에 선다. 또한, 모든 크루는 잠을 자야 하므로 23:59에 집에 돌아간다. 따라서 어떤 크루도 다음날 셔틀을 타는 일은 없다.

입력 형식

셔틀 운행 횟수 n, 셔틀 운행 간격 t, 한 셔틀에 탈 수 있는 최대 크루 수 m, 크루가 대기열에 도착하는 시각을 모은 배열 timetable이 입력으로 주어진다.

• 0 ＜ n ≦ 10
• 0 ＜ t ≦ 60
• 0 ＜ m ≦ 45
• timetable은 최소 길이 1이고 최대 길이 2000인 배열로, 하루 동안 크루가 대기열에 도착하는 시각이 HH:MM 형식으로 이루어져 있다.
• 크루의 도착 시각 HH:MM은 00:01에서 23:59 사이이다.

출력 형식

콘이 무사히 셔틀을 타고 사무실로 갈 수 있는 제일 늦은 도착 시각을 출력한다. 도착 시각은 HH:MM 형식이며, 00:00에서 23:59 사이의 값이 될 수 있다.

입출력 예제

n	t	m	timetable	answer
1	1	5	["08:00", "08:01", "08:02", "08:03"]	"09:00"
2	10	2	["09:10", "09:09", "08:00"]	"09:09"
2	1	2	["09:00", "09:00", "09:00", "09:00"]	"08:59"
1	1	5	["00:01", "00:01", "00:01", "00:01", "00:01"]	"00:00"
1	1	1	["23:59"]	"09:00"
10	60	45	["23:59","23:59", "23:59", "23:59", "23:59", "23:59", "23:59", "23:59", "23:59", "23:59", "23:59", "23:59", "23:59", "23:59", "23:59", "23:59"]	"18:00"

풀이

처음에 문제를 잘못 이해해서, 모든 버스들에 대해 콘이 탈 수 있는지를 구해서 그 중에 가장 늦은 버스를 찾으려고 했다. 그런데 채점을 해보니 5,4,24번 테스트케이스가 틀려서 질문하기를 참고했다.

질문하기의 여러 글 중에서, 이 글을 보고 내 접근방식이 완전히 틀렸다는 것을 깨달았다.

 

콘이 무조건 마지막 버스에 타게 해야한다! 라는 생각으로 문제를 풀었어야 했기 때문이다.

해당 질문하기 글을 보고 풀이를 수정한 결과, 더 짧은 코드로 정답을 얻을 수 있었다.

 

☆핵심★은,

콘을 제외하고

마지막 버스에 탄 인원이 m명보다 작으면, 콘은 마지막 버스 출발 시간에 도착하면 된다.

하지만 마지막 버스에 탄 인원이 m명이면(만원이면), 콘은 마지막 버스의 마지막 탑승객보다 1분 일찍 도착하면 된다.

코드

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

using namespace std;

string minToStr(int min) {
    int h = min/60;
    int m = min%60;
    string result = "";
    if (h<10) result+="0";
    result+=to_string(h);
    result+=":";
    if (m<10) result+="0";
    result+=to_string(m);
    return result;
}
string solution(int n, int t, int m, vector<string> timetable) {
    string answer = "";
    int answerM;
    vector<int> timetableM;
    for (auto item: timetable) {
        int temp = 0;
        temp += stoi(item.substr(0,2))*60;
        temp += stoi(item.substr(3,2));
        timetableM.push_back(temp);
    }
    sort(timetableM.begin(), timetableM.end());
    int num = timetable.size();
    int busT, peo;
    int idx = 0;
    for(int i=0;i<n;i++) {
        busT = 540 + t*i;
        peo = 0;
        while(peo < m && idx < num && timetableM[idx] <= busT) {
            idx++;
            peo++;
        }
    }
    if   (peo < m) answerM = 540 + t*(n-1);
    else answerM = timetableM[idx-1] -1;
    answer = minToStr(answerM);
    return answer;
}

결과