[프로그래머스] [1차] 셔틀버스 / Java
문제주소 :programmers.co.kr/learn/courses/30/lessons/17678
코딩테스트 연습 - [1차] 셔틀버스
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"
programmers.co.kr
<문제 설명>
문제 설명
셔틀버스
카카오에서는 무료 셔틀버스를 운행하기 때문에 판교역에서 편하게 사무실로 올 수 있다. 카카오의 직원은 서로를 '크루'라고 부르는데, 아침마다 많은 크루들이 이 셔틀을 이용하여 출근한다.
이 문제에서는 편의를 위해 셔틀은 다음과 같은 규칙으로 운행한다고 가정하자.
- 셔틀은 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" |
<풀이법>
▒ 한줄 개념: 해시맵, 스택, 큐 ▒
특별한 알고리즘이 필요한 문제는 아니었다.
n, t, m의 사이즈가 충분히 작고, 효율성 테스트 또한 존재하지 않아 기본적인 자료구조와 문자열 슬라이싱을 이용해 풀면 되는 문제이다. 물론 이 단순한 방법을 생각해내기까지 꽤나 오랜시간이 걸렸다..
나는 해시맵을 가장 큰 자료구조로 사용하여 풀었고, 부분적으로 큐와 스택을 사용하였다.
또한, 모든 시간문자열은 분단위를 기준으로 정수로 바꿔주었다. 이렇게 하는 편이 연산에 있어 훨씬 단순하다.
해시맵의 형태는 {셔틀의 출발 시간: 그 셔틀에 탈 수 있는 사람들의 도착 시간} 이다.
540부터 n, t 값을 기준으로 해시맵의 key값을 만들고, 기본 value는 모두 stack으로 만든다.
그 이후, 정수형의 시간 값들(timetable)을 반복문을 통해서 해시맵에 삽입해주는데, 키값보다 작은 값만 채워준다. 또한, 각 해시맵의 value의 크기는 m보다 같거나 작아야한다.
| 1 | 1 | 5 | ["08:00", "08:01", "08:02", "08:03"] |
위와 같은 예시에서, 해시맵은 {540: {480, 481, 482, 483}} 의 형태를 가질 것이다.
| 2 | 10 | 2 | ["09:10", "09:09", "08:00"] |
위와 같은 예시에서, 해시맵은 {540: {480}, 550:{549, 550}}의 형태를 가질 것이다.
이런 식으로 해시맵을 다 채워주고 나면, 해시맵의 가장 큰 키값에 대한 value만 확인한다. 셔틀을 탈 수 있는 가장 마지막 시간을 정답으로 선택해야하므로, 그 앞의 시간들은 상관없다. 가장 큰 키값, 즉 가장 늦은 시간의 셔틀에 탈 수 있는 경우만 확인하면 된다.
따라서, 마지막 셔틀의 시간을 키값으로 둔 value 값(스택)을 확인하는데,
1) 만약 이 stack의 사이즈가 m보다 작다면, 해당 stack에 들어갈 수 있는 가장 큰 값인 셔틀의 시간이 정답이 된다.
| 1 | 1 | 5 | ["08:00", "08:01", "08:02", "08:03"] |
위 예시가 해당 경우인데, 해시맵이 {540: {480, 481, 482, 483}}와 같다. 셔틀에 탄 사람들의 수가 m보다 작으므로, 셔틀이 출발하기 전에만 가면 이 셔틀을 탈 수 있다. 5명이 탈 수 있는 셔틀에 어차피 사람이 4명밖에 없으므로, 484분에 가던, 485분에 가던, 486분에 가던, 540분 전에만 가면 셔틀을 탈 수 있는 것이다. 따라서 가장 늦은 시간인 540분이 정답이 된다.
2) 만약 이 stack의 사이즈가 m과 같다면, 해당 stack에 들어있는 가장 늦게 온 사람의 시간-1 이 정답이 된다.
| 2 | 10 | 2 | ["09:10", "09:09", "08:00"] |
위 예시가 해당 경우인데, 해시맵의 형태는 {540: {480}, 550:{549, 550}}이다.
가장 늦은 셔틀은 550분의 셔틀인데, 이는 2명(m값)이 꽉 차서 빈 자리가 없다. 따라서 마지막 탑승하는 사람보다 빨리가서 그 사람을 밀어내고 자신이 탑승해야하는 것이다. 마지막 탑승하는 사람은 550분에 탑승하러 오므로, 그보다 빠른 시간인 549분에 가면 셔틀에 탑승할 수 있고, 이 시간이 출근할 수 있는 가장 늦은 시간이 된다.
<코드(Java)>
import java.util.*;
class Solution {
HashMap<Integer, Stack<Integer>> map;
Queue<Integer> queue;
int[] keys;
int[] times;
public String solution(int n, int t, int m, String[] timetable) {
map = new HashMap<>();
queue = new LinkedList<>();
keys = new int[n];
times = new int[timetable.length];
for(int i = 0; i < times.length; i++){
times[i] = changeTimeToInt(timetable[i]);
}
Arrays.sort(times);
for(int time: times){
queue.add(time);
}
for(int i = 0; i < n; i++){
map.put(540 + t * i, new Stack<>());
keys[i] = 540 + t * i;
}
int key_cnt = 0;
while (key_cnt < n){
if(queue.isEmpty())
break;
if(keys[key_cnt] >= queue.peek() && map.get(keys[key_cnt]).size() < m){
map.get(keys[key_cnt]).push(queue.poll());
}else{
key_cnt++;
}
}
if(map.get(keys[keys.length-1]).size() < m){
return changeTimeToString(keys[keys.length-1]);
}
else{
return changeTimeToString(map.get(keys[keys.length-1]).peek()-1);
}
}
int changeTimeToInt(String time){
return Integer.parseInt(time.substring(0,2)) * 60 + Integer.parseInt(time.substring(3,5));
}
String changeTimeToString(int time){
String hours = Integer.toString(time/60);
String minutes = Integer.toString(time%60);
hours = hours.length() < 2 ? "0"+hours : hours;
minutes = minutes.length() < 2 ? "0"+minutes : minutes;
return hours+":"+minutes;
}
}
더 많은 코드 보기(GitHub) : github.com/dwkim-97/CodingTest