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

미숙한 블로그 주인이 코딩테스트 문제를 풀어가는 과정을 담은 글입니다. 이 풀이가 효율적인 풀이가 아닐 수 있으며, 부정확한 정보가 많이 있을 수 있습니다. 보완해야할 점이 있다면 댓글로 남겨주세요!

 

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

프로그래머스

[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 사이의 값이 될 수 있다.

 

풀이

입출력 예시

 

풀이

먼저 이 문제는 두 가지 경우를 나눠서 생각해야한다.

 

첫 번째는, 마지막 셔틀버스에 자리가 남는 경우이다. 이 경우에는 콘은 마지막 셔틀이 오는 시간에 맞춰서 대기하고 있으면 된다.

 

생각이 필요한 것이 두 번째 상황인데, 마지막 셔틀버스에 자리가 남지 않는 경우이다. 예를 들어서 보자.

마지막 셔틀버스가 9:00에 도착하고, 2명의 크루만 탈 수 있다. 이 때의 경우가 또 두 가지로 나뉜다.

1. 크루가 각각 [8:59, 9:00] 에 도착했을 때.

이 경우에는 콘은 8:59에 도착해야한다. 그러면 줄을 선 크루는 [8:59, 8:59(콘), 9:00] 이 되므로 콘은 셔틀버스를 탈 수 있다.

 

2. 크루가 각각 [8:59, 8:59] 에 도착했을 때.

콘은 항상 자기와 같은 시간에 온 크루가 있다면 가장 마지막에 줄을 선다. 그렇기 때문에 8:59에 줄을 서게 된다면 콘은 셔틀 버스를 타지 못한다. 그래서 8:58분에 나와야 셔틀버스를 탈 수 있다.

 

이 경우들을 생각하면서 문제를 풀어나갔다.

 

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

using namespace std;

string solution(int n, int t, int m, vector<string> timetable) {
    string answer = "";
    vector<int> t_time;
    
    // 계산이 쉽도록 시간을 분으로 표시
    for (string time : timetable) {
        string hour = time.substr(0, 2);
        string min = time.substr(3, 2);
        t_time.push_back(stoi(hour) * 60 + stoi(min));
    }
    
    sort(t_time.begin(), t_time.end());
    
    // 첫 차 시간은 09:00. 즉, 540분.
    int shuttle_bus = 540;
    int idx = 0;
    int t_size = t_time.size();
    int last_time;
    
    for (int i = 0; i < n; i++) {
        int count = 0;
        // 크루 한 명씩 태우기
        for (int j = idx; j < t_size; j++) {
            if (t_time[j] <= shuttle_bus) {
                idx++;
                count++;
                // 꽉 차면 다음 버스
                if (count == m) {
                    break;
                }
            }
        }
        
        // 마지막 버스
        if (i == n - 1) {
            // 버스가 가득 차면?
            if (count == m) {
                last_time = t_time[idx - 1] - 1;
            }
            // 버스에 자리가 남으면
            else {
                last_time = shuttle_bus;
            }
        }
        
        shuttle_bus += t;
    }

    int hour = last_time / 60;
    int min = last_time % 60;
    
    if (hour < 10) {
        answer = "0" + to_string(hour) + ":";
    }
    else {
        answer = to_string(hour) + ":";
    }
    
    if (min < 10) {
        answer = answer + "0" + to_string(min);
    }
    else {
        answer = answer + to_string(min);
    }
    
    return answer;
}