프로그래머스 level3 [ [1차] 셔틀버스] 파이썬

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

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

# 프로그래머스 [1차] 셔틀버스

from collections import deque


def solution(n, t, m, timetable):
    answer = ''
    bus_time = []
    bus_cnt = [[] for _ in range(n)]
    for i in range(n):
        bus_time.append(540+i*t)
    for i in range(len(timetable)):
        a, b = map(int, timetable[i].split(':'))
        timetable[i] = a*60+b
    timetable.sort()
    i = 0
    j = 0
    cnt = 0
    while i < len(timetable) and j < n:
        if timetable[i] <= bus_time[j]:
            cnt += 1
            bus_cnt[j].append(timetable[i])
            if cnt == m:
                cnt = 0
                j += 1
        else:
            while j < n:
                if timetable[i] <= bus_time[j]:
                    break
                j += 1
            cnt = 1
            if j < n:
                bus_cnt[j].append(timetable[i])
        i += 1
    if len(bus_cnt[n-1]) != m:
        answer = bus_time[n-1]
    else:
        answer = bus_cnt[n-1][m-1]-1
    H = answer//60
    M = answer % 60
    answer = str(H).rjust(2, '0')+':'+str(M).rjust(2, '0')
    return answer

bus_time에 각 버스의 시간을 분으로 변환하여 저장한다. 

timetable에 각 사람의 정류장 도착시간을 분으로 변환하여 저장한다.

bus_cnt에는 각 시간 버스당 태울 수 있는 사람의 정보 (정류장 도착시간)을 저장한다.

 

각 버스에 도착하는 사람을 순서대로 버스에 태우고, 다 차면 다음 버스를 태운다.