[python]프로그래머스(level3) - 셔틀버스

울상냥·2022년 8월 29일

programmers python

Programmers

목록 보기

10/22

0. 문제설명

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

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

셔틀은 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 사이의 값이 될 수 있다.

1. code

from datetime import timedelta, datetime

def solution(n, t, m, timetable):
    
    fmt = "%H:%M" #시간 형식
    term = timedelta(minutes = t) #셔틀 간격
    
    timetable.sort(reverse=True)
    shuttles = [] #셔틀 시간표
    shuttle = datetime(2022,8,25,9,0,0) #첫번째 셔틀

    #셔틀 시간표 생성
    for i in range(0, n):
        shuttles.append(shuttle.strftime(fmt))
        shuttle = shuttle + term
    
    #셔틀 별 탑승크루 파악
    for s in shuttles:
        crew = [] #셔틀 별 탑승크루
        while(len(crew) < m): #m명까지 탑승가능
            if timetable:
                if timetable[-1] <= s: #셔틀도착 시간까지 대기열에 선 경우
                    crew.append(timetable.pop())
                else:
                    break
            else:
                break

    #마지막 셔틀에서 
    if len(crew) == m: #탑승인원이 가득 찬 경우
        last_time = datetime.strptime(crew[-1], fmt) - timedelta(minutes=1) #가장 마지막에 도착한 크루보다 1분 일찍 도착
        answer = last_time.strftime(fmt)
    else: #탑승인원이 가득 차지 않은 경우
        answer = shuttles[-1] #셔틀 도착 시간에 맞춰 도착
    
    return answer

2. 문제풀이

자료구조

shuttles = [셔틀 도착 시간]
crew = [각 셔틀의 탑승크루 도착 시간]

설명

첫 번째 셔틀인 9시 셔틀을 시작으로 term간격으로 하여 shuttles시간표를 생성한다.
shuttles의 첫 번째 셔틀부터 탑승 인원을 파악한다.
마지막 셔틀의 탑승인원인 crew에서 탑승 인원이 가득 찬 경우 가장 마지막에 도착한 크루보다 1분 빠르게 도착해야하고 탑승 인원이 가득 차지 않은 경우 셔틀 도착 시간에 맞춰 도착하도록 한다.