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

문제설명

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

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

• 셔틀은 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 사이이다.

전체코드(Python)

# 출발시간과 crew의 도착 시간이 동일하거나 이전이면 True 이후면 False
def compare_time(crew,departure):
    if int(departure[:2])>int(crew[:2]):
        return True
    elif int(departure[:2]) == int(crew[:2]):
        if int(departure[3:])>= int(crew[3:]):
            return True
    return False

# 시간을 빼고 더하는 함수
def cal_time(time,interval,operator):
    # True일 경우 시간을 더한다.
    if operator:
        if int(time[3:])+interval >= 60:
            time = str(int(time[:2])+1).rjust(2,'0')+":"+str((int(time[3:])+interval)%60).rjust(2,'0')
        else:
            time = str(int(time[:2])).rjust(2,'0')+":"+str((int(time[3:])+interval)).rjust(2,'0')
    # False일 경우 시간을 뺀다.
    else:
        if int(time[3:])-interval < 0:
            time = str(int(time[:2])-1).rjust(2,'0')+":"+str((int(time[3:])+60-interval)%60).rjust(2,'0')
        else:
            time = str(int(time[:2])).rjust(2,'0')+":"+str((int(time[3:])-interval)).rjust(2,'0')
    return time
def solution(n, t, m, timetable):
    answer = ''
    # 크루들을 정류장에 온 시간순으로 배치한다.
    timetable = sorted(timetable)
    # 마지막 차가 오기 전까지 크루들을 셔틀에 태워서 보낸다.
    time = "09:00"
    while n>1:
        i = m
        # 크루가 셔틀버스가 출발하기 전에 도착해 있었거나, 자리가 있다면 태운다.
        while compare_time(timetable[0],time) and i>0 and len(timetable)>0:
            # 한명의 크루를 셔틀에 태운다.
            timetable.pop(0)
            i-=1
        # 다음 버스 운행시간으로 변경한다.    
        time = cal_time(time,t,True)
        n-=1
    
    # 이번 버스에서 탈 수 있는 인원의 수를 구한다.
    person = 0
    for i in range(len(timetable)):
        if compare_time(timetable[i],time):
            person+=1
    
    # 탈 수 있는 인원이 m을 초과하는경우
    if person>=m:
        # m번째 인원보다 1분 앞서서 오면 된다.
        last = cal_time(timetable[m-1],1,False)
    else:
    # 타야하는 인원이 m을 초과하지 않는경우 버스 출발 시간에 오면 된다.
        last = time
    
    return last

알아야할 함수들

• departure[:2] : 슬라이싱
  • 문자열을 [시작 인덱스:마지막 인덱스+1]만큼 슬라이스 하여 반환한다.
• str(int(time[:2])+1).rjust(2,'0') : 길이만큼 문자열 채우기
  • 문자열을 정해진 길이만큼 정해진 문자로 채우는 함수이다.
  • 위와 유사한 함수로
    • 문자열.zfill(길이) : 왼쪽부터 0을 채운다. 
    • 문자열.ljust(길이,채울문자) : 오른쪽부터 길이만큼 채울문자를 채운다.
• sorted(timetable) : 정렬
  • 배열을 정해진 규칙에 따라서 아이템들을 정렬한다.

 

혹시라도 정정할 내용이나 추가적으로 필요하신 정보가 있다면 댓글 남겨주시면 감사하겠습니다.

오늘도 Jindory 블로그에 방문해주셔서 감사합니다.

 

[참조]

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