[프로그래머스] 셔틀 버스 / 파이썬 / (시뮬레이션)

https://programmers.co.kr/learn/courses/30/lessons/17678?language=python3

코딩테스트 연습 - [1차] 셔틀버스

풀이

은근히 까다로웠던 문제이다.
푼 방법으로는 우선, 시간을 모두 분 단위로 변환한 후
모든 크루원을 한 명씩 버스에 태웠다.
한 명씩 버스에 태우다보면, 마지막 버스에, 마지막 사람이 언제 탔는지를 알 수 있는데 그 값의 -1을 한 시간을 문자열로 반환하면 된다.
만약 마지막 버스에 전부 타지 않았다면, 마지막 버스 시간을 리턴하면 된다.

def solution(n, t, m, timetable):
    answer = ''
    crew = [ int(tt.split(":")[0])*60+int(tt.split(":")[1]) for tt in timetable ]
    # 크루의 도착시간
    crew.sort()

    # bus[x] = (버스시간, 승객 수, 마지막 탄 크루의 도착시간)
    bus = [ (540+t*i, 0, None) for i in range(n)]

    bidx, cidx = 0, 0
    while cidx < len(crew):
        c = crew[cidx]
        if bidx == len(bus):
            break
        if c <= bus[bidx][0] and bus[bidx][1] < m:
            tt, cnt, _ = bus[bidx]
            bus[bidx] = (tt, cnt+1, c)
            cidx += 1
        else:
            bidx += 1

    ret = bus[-1][0]
    if bus[-1][2]:
        if bus[-1][1] == m:
            ret = bus[-1][2] - 1

    return '%02d:%02d' % (ret // 60, ret % 60)