카카오 - 셔틀버스

아따맘마·2021년 1월 8일
0

알고리즘 - 카카오

목록 보기
4/19

문제

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

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

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

풀이

아직 알고리즘 하수다 보니 이문제를 푸는데 너무 오랜 시간이 걸렸다...
이 문제를 푸는데 핵심은 이것이다.

  • 크루 timetable을 정수형으로 바꿔 순서대로 정렬하기
    string을 정수형으로 바꾸는건 stoi함수를 사용해서 변환해줬다. 그리고 vector변수에다 담아서 sort했다.
  • 마지막 버스가 만원인지 아닌지 판단하기
    여기서 진짜 애를 먹었다... 계속 vector의 범위가 벗어난 곳을 참조해서 오류가 뜨느라 수정하는데만 한시간?걸린듯 하다.
for (full = 0; full < m; full++) {
	if (person < timetable.size() && time[person] <= busTime)
		person++;
	else if (person == timetable.size() || time[person] > busTime)
		break;

이 부분에서 벡터 범위 오류가 계속 발생했다. 그래서 조건문 순서를 이용해서 먼저 person이 timetable.size()보다 작은지 판단하고 다음 조건을 보는 식으로 수정했다.
그리고 만약 person이 timetable.size()와 같거나 time[person]이 버스 정차 시간보다 크면 반복문을 탈출!. time[person]이 버스 정차 시간보다 크다는 것은 이 크루부터는 다 못타기 때문이다.(오름차순 정렬을 했으므로)

  • 만원 조건
    문제에서 주어진 테스트 케이스는 다 통과해서 제출했더니 반만 맞았더라... 그래서 또 뭐가 잘못된건지 살펴보는데..
if (i == n && full == m && person <= timetable.size())
	lastFull = true;

이 부분에서

person == timetable.size()

로 했었었다. 나는 맨 처음에 모든 person을 다 버스 정차 시간과 비교할 줄 알았는데 아니었다.
생각해보니 다음과 같은 케이스에서는 일부만 조건문으로 탐색한다.

마지막 버스 정차 시간 : 09:10
남은 사람 3명
사람1 : 09:10
사람2 : 09:20
사람3 : 09:30
이런 케이스에서는 사람3은 애초에 탐색을 안해도 된다.

그래서 마지막 버스 정차 시간(i == n)에 사람은 꽉 찼지만(full == m) person은 timestable.size()보다 작거나 같을 수도 있다.

코드

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

using namespace std;

string solution(int n, int t, int m, vector<string> timetable) {
	string answer = "";
	int busTime = 540;
	int person = 0;
	int full;
	bool lastFull = false;
	int tmp;
	vector<int> time(timetable.size());
	for (int i = 0; i < timetable.size(); i++)
		time[i] = stoi(timetable[i].substr(0, 2)) * 60 + stoi(timetable[i].substr(3));
	sort(time.begin(), time.end());
	for (int i = 1; i <= n; i++)
	{
		for (full = 0; full < m; full++) {
			if (person < timetable.size() && time[person] <= busTime)
				person++;
			else if (person == timetable.size() || time[person] > busTime)
				break;
		}
		if (i == n && full == m && person <= timetable.size())
			lastFull = true;
		if (i != n)
			busTime += t;
	}
	if (lastFull)
		tmp = time[person - 1] - 1;
	else
		tmp = busTime;
	
	//시간 변경
	if (tmp / 60 < 10)
	{
		if (tmp % 60 < 10)
			answer = "0" + to_string(tmp / 60) + ":0" + to_string(tmp % 60);
		else
			answer = "0" + to_string(tmp / 60) + ":" + to_string(tmp % 60);
	}
	else
	{
		if (tmp % 60 < 10)
			answer = to_string(tmp / 60) + ":0" + to_string(tmp % 60);
		else
			answer = to_string(tmp / 60) + ":" + to_string(tmp % 60);
	}
		return answer;
}

// 테스트 케이스
int main()
{
	vector<string> timetable1 = { "08:00" , "08:01", "08:02", "08:03"};
	vector<string> timetable2 = {"09:10", "09:09", "08:00" };
	vector<string> timetable3 = { "09:00", "09:00", "09:00", "09:00" };
	vector<string> timetable4 = { "00:01", "00:01", "00:01", "00:01", "00:01" };
	vector<string> timetable5 = { "23:59"};
	vector<string> timetable6 = { "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" };
	cout << "1 : " << solution(1, 1, 5, timetable1) << '\n';
	cout << "2 : " << solution(2, 10, 2, timetable2) << '\n';
	cout << "3 : " << solution(2, 1, 2, timetable3) << '\n';
	cout << "4 : " << solution(1, 1, 5, timetable4) << '\n';
	cout << "5 : " << solution(1, 1, 1, timetable5) << '\n';
	cout << "5 : " << solution(10, 60, 45, timetable6) << '\n';

}
profile
늦게 출발했지만 꾸준히 달려서 도착지점에 무사히 도달하자

0개의 댓글