[2018 카카오 블라인드 채용] 셔틀버스

문제

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

프로그래머스

 

소스코드

import java.util.Arrays;
import java.util.LinkedList;
import java.util.List;
import java.util.PriorityQueue;
import java.util.stream.Collectors;

class Solution {
    public String solution(int n, int t, int m, String[] timetable) {

        List<String> busTimeTable = makeBusTimeTable(n, t);
        String lastBusTime = busTimeTable.get(n-1);
        PriorityQueue<String> crew = Arrays.stream(timetable)
                                        .collect(Collectors.toCollection(PriorityQueue::new));

        for(int i=0; i<n; ++i) {
            String busTime = busTimeTable.get(i);
            for(int j=0; j<((i < n-1)? m : (m-1)); ++j) {
                if(crew.isEmpty()) return lastBusTime;
                if(crew.peek().compareTo(busTime)>0) break;
                crew.poll();
            }
        }

        if(crew.isEmpty()) return lastBusTime;

        String time = minusOneMin(crew.poll());
        return (lastBusTime.compareTo(time)<0)? lastBusTime : time;
    }

    public List<String> makeBusTimeTable(int n, int t) {

        List<String> busTimeTable = new LinkedList<>();
        int h = 9, m = 0;

        for(int i=0; i<n; ++i) {
            busTimeTable.add(String.format("%02d:%02d", h, m));
            if((m += t) >= 60) {
                h++;
                m -= 60;
            }
        }

        return busTimeTable;
    }

    public String minusOneMin(String time) {

        String[] tmp = time.split(":");
        int h = Integer.parseInt(tmp[0]);
        int m = Integer.parseInt(tmp[1]);

        if(m > 0) {
            m -= 1;
        } else { 
            h -= 1; 
            m = 59; 
        }

        return String.format("%02d:%02d", h, m);
    }
}

 

회고

 - 카카오가 이런 시간 문자열 문제를 좋아하는 듯.

 - 처음에는 버스 시간 t마다 무조건 수용인원 m만큼 크루들을 빼줬는데, 버스 시간마다 도착해 있는 크루의 수가 다르기 때문에 버스 시간표를 만들어서 해당 시간에 줄 서있는 수 만큼씩 빼줌.