수학, 알고리즘

[프로그래머스 Lv3] 셔틀버스

춤추는수달 2022. 7. 21. 15:42

문제 설명

셔틀버스

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

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

  • 셔틀은 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 사이의 값이 될 수 있다.

입출력 예제

ntmtimetableanswer
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"

풀이

단순히 셔틀에 사람들을 다 태우고, 마지막 셔틀에 빈 자리가 있으면 정답을 '마지막 셔틀의 도착시간' 으로 하고,

마지막 셔틀에 빈 자리가 없으면 정답을 '가장 늦게 탄 사람보다 1분 일찍'으로 하면 된다.

딱히 알고리즘 같은건 필요하지 않다. 거슬리는 부분은 시간비교를 위해  string과 int 변환 정도. 나는 시/분을 따로 계산했는데, 다른 사람의 풀이를 보니 시간 계산 시 시를 분으로 합쳐서 계산했다. 그게 더 간편하고 좋아보였다. 

코드

#include <string>
#include <vector>
#include <iostream>
#include <queue>

using namespace std;

int toNum(string s)
{
    int ret = 0;
    ret += (s[0]-'0')*1000;
    ret += (s[1]-'0')*100;
    ret += (s[3]-'0')*10;
    ret += (s[4]-'0');
    return ret;
}
string toTime(int n)
{
    string ret;
    string hour;
    string min;
    if(n/100 < 10) hour += '0';
    hour += to_string(n/100);
    if(n%100 < 10) min += '0';
    min += to_string(n%100);
    
    ret += hour;
    ret += ':';
    ret += min;
    cout << "n :"<<n <<" ret: "<<ret<<endl;
    return ret;
}
int timePlus(int a, int b)
{
    int ret;
    a += (b/60) * 100;
    a += b%60;
    return a;
}
int timeMinus(int a, int b)
{
    int ret = 0;

    int beforeH = a/100;
    int beforeM = a%100;
    ret = beforeH*60 + beforeM;
    ret += b;
    ret = 100*(ret/60)+ (ret%60);
    return ret;
}
string solution(int n, int t, int m, vector<string> timetable) {
    string answer = "";
    
    int tableSize = timetable.size();
    priority_queue<int, vector<int>,greater<int>> sortedTable;
    
    for(auto s : timetable)
    {
        sortedTable.push(toNum(s));
    }
    
    vector<vector<int>> pickUp(n);
    int pIndex = 0;
    for(int i = 0; i < n ;i++)
    {
        //i번쨰 셔틀 도착
        int nowShuttle = timePlus(900,i*t);
        int inCar = 0;
        
        cout << i<<"번쨰 셔틀 도착 "<< nowShuttle<<endl;
        //최대 m명까지 태우기
        while(inCar < m && !sortedTable.empty())
        {
            //도착시간이 셔틀 도착시간보다 빨랐으면 태우기
            if(sortedTable.top() <= nowShuttle )
            {
                pickUp[i].push_back(sortedTable.top());
                cout << "승객 탑승 " << sortedTable.top()<<endl;
                sortedTable.pop();
                inCar++;
            }else
            {
                break;
            }
        }
       
    }
    
    for(auto s : pickUp)
    {
        for(auto p : s)
        {
            cout << p <<endl;
        }
    }
    if(pickUp[n-1].size() < m)
    {
        answer = toTime(timePlus(900,(n-1)*t));
    }else
    {
        answer = toTime(timeMinus(pickUp[n-1].back(), -1));
    }
    //timetable순회
    //빨리온 사람부터 셔틀에 태움
    //가장 늦은 셔틀에 빈 자리가 있으면  답 = 가늦셔 도착시간 
    //아니면 셔틀에 탄 사람 중 가장 늦게 온 사람 보다 1분 먼저 도착

    
    return answer;
}