[프로그래머스 Level.3] [1차] 셔틀버스 (2018 KAKAO BLIND RECRUITMENT) (Java)

문제 링크

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

프로그래머스

 

 

 

 

코딩테스트 연습 > 2018 KAKAO BLIND RECRUITMENT > [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"

 

 

 

 

 

 

나의 코드

import java.util.*;

class Solution {
    public String solution(int n, int t, int m, String[] timetable) {
        String answer = "";
        int conTime = 0;
        
        int[] timeArr = new int[timetable.length];
        
        for(int i=0; i<timetable.length; i++) {
            String[] s = timetable[i].split(":");
            timeArr[i] = Integer.parseInt(s[0]) * 60 + Integer.parseInt(s[1]);
        }
        
        Arrays.sort(timeArr);
        
        Queue<Integer> q = new LinkedList<>();
        for(int i=0; i<timeArr.length; i++) {
            q.offer(timeArr[i]);
        }
        
        ArrayList<ArrayList<Integer>> list = new ArrayList<>();
        for(int i=0; i<n; i++) {
            list.add(new ArrayList<>());
        }
        
        int busTime = 540; // 09시
        
        for(int i=0; i<n; i++) {
            while(!q.isEmpty()) {
                int crew = q.peek();
                if(crew <= busTime && list.get(i).size() < m) {
                    list.get(i).add(q.poll());
                    conTime = crew - 1;
                } else {
                    break;
                }
            }
            busTime += t;
        }
        
        if (list.get(n-1).size() < m) conTime = busTime - t;
        
        // format 함수를 사용하여 간편하게 변환 할 수 있다.
        // String hour = String.format("%02d", conTime / 60);
        // String min = String.format("%02d", conTime % 60);
        String hour = (conTime / 60 < 10) ? ("0" + conTime / 60) : ("" + conTime / 60);
        String min = (conTime % 60 < 10) ? ("0" + conTime % 60) : ("" + conTime % 60);
        
        answer = hour + ":" + min;
        
        return answer;
    }
}

 

풀이

1. 주어진 timetable의 각 문자열을 분으로 변경하여 계산하기 편하게 만들어 int형 배열 timeArr에 저장한다.
2. timeArr을 오름차순으로 정렬한다.
3. Queue를 생성하여 timeArr의 데이터를 순서대로 담아 각 크루들이 셔틀을 타러 오는 시간의 순서대로 Queue에 담기도록 한다.
4. ArrayList를 선언타입으로 하는 ArrayList를 생성하여 운행 횟수만큼 ArrayList에 크루들이 탑승하도록 한다.
5. 셔틀은 9시부터 첫 운행을 하므로 int형 변수 busTime을 선언하고 9시를 분으로 고친 540으로 초기화한다.
6. 이제 셔틀 운행 횟수인 n만큼 for문을 돌면서 각 운행마다 크루들을 탑승시키며 콘이 탑승하는 시간을 마지막에 탑승한 크루의 탑승 시간 - 1분으로 업데이트 한다.
7. Queue가 빌 때까지 for문을 돌면서 Queue의 첫 데이터(현재 제일 앞에서 대기하고 있는 크루의 탑승 시간)가 busTime 이하이고 현재 셔틀에 크루 수가 꽉 차지 않았다면(list.get(i).size()가 m보다 작다면) 해당 크루를 버스에 탑승시키고 콘이 탑승하는 시간을 해당 크루가 탑승한 시간 - 1분으로 업데이트 한다. (계속해서 크루가 탑승한 시간 - 1분으로 업데이트하면 최종적으로 마지막에 탑승한 크루의 탑승 시간 - 1분으로 되어 마지막 셔틀에 탑승할 수 있다)
8. Queue의 첫 데이터인 크루의 셔틀 탑승 시간이 현재 busTime(셔틀 운행 시간)보다 크거나 셔틀에 탑승 인원이 꽉 찼다면 while문을 종료하고 busTime을 t만큼 증가시키고 다음 셔틀을 운행한다.
9. 이렇게 셔틀 운행 횟수인 n만큼 for문을 돌고 나오면 conTime(콘의 탑승 시간)은 마지막 탑승자 - 1분으로 되어 마지막 셔틀을 탑승할 수 있게 되어있다.
10. 만약 마지막 셔틀에 탑승 인원이 꽉 차있지 않다면(list.get(n-1).size() < m) 콘은 해당 마지막 셔틀 시간에 딱 맞춰오면 되기 때문에 conTime을 마지막 셔틀 시간(busTime - t)으로 한다. (busTime - t인 이유는 for문을 돌면서 마지막에 busTime에 t를 한 번 더 증가시키고 for문을 종료했기 때문이다)
11. 콘의 탑승 시간을 구했으니 해당 시간을 다시 String 타입으로 변경하여 answer에 저장하고 반환한다.