[프로그래머스 Level3] [1차] 셔틀버스 - 파이썬(Python) - 2018 KAKAO BLIND RECRUITMENT

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

프로그래머스

 

문제 설명

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

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

• 셔틀은 09:00부터 총 n회 t분 간격으로 역에 도착하며, 하나의 셔틀에는 최대 m명의 승객이 탈 수 있다.
• 셔틀은 도착했을 때 도착한 순간에 대기열에 선 크루까지 포함해서 대기 순서대로 태우고 바로 출발한다. 예를 들어 09:00에 도착한 셔틀은 자리가 있다면 09:00에 줄을 선 크루도 탈 수 있다.

일찍 나와서 셔틀을 기다리는 것이 귀찮았던 콘은, 일주일간의 집요한 관찰 끝에 어떤 크루가 몇 시에 셔틀 대기열에 도착하는지 알아냈다. 콘이 셔틀을 타고 사무실로 갈 수 있는 도착 시각 중 제일 늦은 시각을 구하여라.

단, 콘은 게으르기 때문에 같은 시각에 도착한 크루 중 대기열에서 제일 뒤에 선다. 또한, 모든 크루는 잠을 자야 하므로 23:59에 집에 돌아간다. 따라서 어떤 크루도 다음날 셔틀을 타는 일은 없다.

 

---

 풀이 과정

1. "HH:MM"형태의 크루들의 도착시간을 모두 분으로 바꾼 후 오름차순 정렬해준다.

2. 마지막 버스 제외 버스들의 도착시간에 탈 수 있는 사람들은 모두 버스를 탄다.

3. 마지막 버스가 왔을 때 탈 수 있는 사람의 수가 m명 이상이라면 콘은 (timetable[tableIdx+m-1] - 1)의 시간에 줄을 선다.

 - 콘이 없었다면 마지막으로 버스를 탔었을 사람보다는 최소 1분 빨리 줄을 서야한다.

4. 마지막 버스가 왔을 때 탈 수 있는 사람의 수가 m명 미만이라면 콘은 마지막 버스 시간에 줄을 선다.

 

 

<< 전체 코드 >>

def change_str_to_minutes(s): # 문자열을 분으로 바꾸는 함수
    return int(s[:2]) * 60 + int(s[-2:])

def change_minutes_to_str(m): # 분을 문자열로 바꾸는 함수
    Time = [str(m // 60), str(m % 60)]
    for i in range(2):
        if len(Time[i]) == 1:
            Time[i] = '0' + Time[i]
    return Time[0] + ':' + Time[1]

def solution(n, t, m, timetable):
    for i in range(len(timetable)):
        timetable[i] = change_str_to_minutes(timetable[i])
    timetable.sort()
    
    tableIdx = 0
    bus_arrivalTime = 540
    for _ in range(n-1):
        count = 0  # 마지막을 제외한 버스들의 도착 시간에 탈 수 있는 사람은 버스를 탄다.
        while tableIdx < len(timetable) and count < m and timetable[tableIdx] <= bus_arrivalTime:
            tableIdx += 1
            count += 1
        bus_arrivalTime += t
    
    # 마지막 버스에서 콘은 자리 여유가 없으니 마지막 사람보다 1분 일찍 줄을 선다.
    if tableIdx+m-1 < len(timetable) and timetable[tableIdx+m-1] <= bus_arrivalTime:
        return change_minutes_to_str(timetable[tableIdx+m-1] - 1)
    # 마지막 버스에서 콘은 자리 여유가 있으니, 버스 시간 정각에 줄을 선다.
    return change_minutes_to_str(bus_arrivalTime)