[프로그래머스] 셔틀버스 (Python)

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

프로그래머스

Status : Solved

Time : 00:22:38

 

---

 

 

문제 설명

 

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

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

• 셔틀은 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	result
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"

 

 

---

 

 

풀이

 

카카오 코테, 그리고 LV3문제중에서도 어렵지 않은 문제였던 것 같다(사람마다 개인차가 있으니 주의!)

버스가 한 번에 올때, 승객들 중 "이전 버스에 타지 못했으며, 현재 버스에 타기 전 시간에 도착했으며, 정원수 m 안에 들 수 있는 사람"이 탑승이 가능하게 된다. 따라서 승객을 정렬 후 큐로 취급하면 문제 풀이가 쉬워진다.

우리의 목표는 콘이 사무실에 최대한 늦게 버스를 탈 수 있게 도착 시간을 정하는 것이므로, 각 버스 도착 시간마다 다음과 같은 경우를 따져가며 최대 시간을 갱신해나가면 된다.

• 현재 탑승 가능한 승객 수가 버스 정원에 달할 경우 : 콘은 그냥 버스 탑승 시간에 도착하면 된다.
• 버스 정원을 넘을 경우 : 콘이 마지막으로 탑승할 수 있는 탑승객보다 1분 빠르게 도착하면 된다(콘은 같은 시간에 도착한 사람들보다 우선순위가 낮다)

그리고 이 때의 콘의 도착 시간을 저장하고, 현재 탑승 가능한 인원만큼을 큐에서 제거한 후 다음 버스 도착 시간으로 이동하면 된다.

• time_convert, rev_time_convert : 모든 시간 데이터는 시와 분이 나뉜 문자열 형태로 제공되므로, 원활한 정렬 및 대소비교 연산을 위해서 정수형과의 변환이 필요하다. 이를 담당하는 함수가 위 둘이 된다.
• solution : 앞서 설명한 알고리즘을 토대로 문제 풀이를 수행한다.

 

풀이 코드

def time_convert(t) :
    h, m = map(int, t.split(':'))
    return h*60 + m

def rev_time_convert(t) :
    h, m = t // 60, t % 60
    return '{:02d}:{:02d}'.format(h,m)

def solution(n, t, m, timetable):
    bustable = [ 540 + x*t for x in range(n) ]
    timetable = sorted(list(map(time_convert, timetable)))
    answer = 0
    
    while bustable :
        current_bus = bustable.pop(0)
        cnt = 0
        while cnt < len(timetable) :
            if timetable[cnt] > current_bus :
                break
            cnt += 1
        
        bus_limit = min(cnt, m, len(timetable))
        if cnt < m :
            answer = max(answer, current_bus)
        else :
            answer = max(answer, timetable[bus_limit-1]-1)
        
        for _ in range(bus_limit) :
            timetable.pop(0)
    
    return rev_time_convert(answer)

 

 지금 프로그래머스 접속이 원활하지 않다 보니, 풀이 완료 인증은 뒤로 미뤄야 할 것 같다...