Programmers / [1차] 셔틀버스 / Python 파이썬

 

*문제 출처는 프로그래머스에 있습니다.

문제 제목: [1차] 셔틀버스 - 3단계

문제 사이트: https://school.programmers.co.kr/learn/courses/30/lessons/17678#qna

프로그래머스

문제 설명

셔틀버스

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

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

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

입출력 예제

ntmtimetableanswer

1	1	5	["08:00", "08:01", "08:02", "08:03"]	"09:00"
2	10	2	["09:10", "09:09", "08:00"]	"09:09"
2	1	2	["09:00", "09:00", "09:00", "09:00"]	"08:59"
1	1	5	["00:01", "00:01", "00:01", "00:01", "00:01"]	"00:00"
1	1	1	["23:59"]	"09:00"
10	60	45	["23:59","23:59", "23:59", "23:59", "23:59", "23:59", "23:59", "23:59", "23:59", "23:59", "23:59", "23:59", "23:59", "23:59", "23:59", "23:59"]	"18:00"

---

나의 풀이

 

이 문제는 조건만 잘 생각하면 쉽게 풀린다.

23:59 에는 버스를 타는게 아니라 그냥 걸어서 집에 가는거라고한다.. (문제 조건이 조금 이상함)

 

고려해야할 조건들

 

# 09:00부터 총 n회 t분 간격으로 역에 도착하며, 하나의 셔틀에는 최대 m명의 승객이 탈 수 있다.
# 콘은 게으르기 때문에 같은 시각에 도착한 크루 중 대기열에서 제일 뒤에 선다.
# 모든 크루는 잠을 자야 하므로 23:59에 집에 돌아간다.
# 셔틀 운행 횟수 n, 셔틀 운행 간격 t, 한 셔틀에 탈 수 있는 최대 크루 수 m
# 콘이 무사히 셔틀을 타고 사무실로 갈 수 있는 제일 늦은 도착 시각을 출력한다. 

# 막차가 도착한 시간에 대기열에 있는 사람이, m보다 작으면 콘은 버스도착시간에 맞춰 도착함.
# m보다 클경우, 마지막 순번보다, 1분 더빨리 도착해버림.
# 콘이 막차를 못타는 테스트케이스는 존재도 하지 않음

# 09:00부터 총 n회 t분 간격으로 역에 도착하며, 하나의 셔틀에는 최대 m명의 승객이 탈 수 있다.
# 콘은 게으르기 때문에 같은 시각에 도착한 크루 중 대기열에서 제일 뒤에 선다.
# 모든 크루는 잠을 자야 하므로 23:59에 집에 돌아간다.
# 셔틀 운행 횟수 n, 셔틀 운행 간격 t, 한 셔틀에 탈 수 있는 최대 크루 수 m
# 콘이 무사히 셔틀을 타고 사무실로 갈 수 있는 제일 늦은 도착 시각을 출력한다. 

# 막차가 도착한 시간에 대기열에 있는 사람이, m보다 작으면 콘은 버스도착시간에 맞춰 도착함.
# m보다 클경우, 마지막 순번보다, 1분 더빨리 도착해버림.
# 콘이 막차를 못타는 테스트케이스는 존재도 하지 않음
def solution(n, t, m, timetable):
    answer = ''
    start_time = 540 # 셔틀차 시작시간
    end_time = 1439 # 모든 크루가 꼭 돌아가야하는 시간
    mintable = [time2min(time) for time in timetable if time2min(time) < 1439]
    mintable.sort()
    
    cnt = 0
    stack = []
    while start_time < 1440:
        times_cut = [i for i in mintable if i <= start_time] # 9시 또는 9시 이전에 도착한 크루
        
        if times_cut:
            if n > 1:
                n -= 1
                mintable = mintable[m:]
                
            elif n == 1:
                if len(times_cut) >= m:
                    return min2time(times_cut[m-1] - 1) 
                else:
                    if times_cut[-1] == start_time:
                        return min2time(times_cut[-1]-1)
                    else:
                        return min2time(start_time)
        else:
            if n > 1:
                n -= 1
            else:
                return min2time(start_time)
        start_time += t 
    
    return min2time(start_time)

def time2min(time):
    h,m = time.split(':')
    return (int(h)*60) + int(m)

def min2time(min):
    time = ""
    h = min // 60
    m = min % 60
    if h < 10:
        time += "0" + str(h)
    else:
        time += str(h)
    if m < 10:
        time += ":0" + str(m)
    else:
        time += ":" + str(m)
    return time

 

min2time 함수를 좀더 간결하게 고치면

def convert_string(m):
    h = m // 60
    m = m % 60
    return '{0:02d}:{1:02d}'.format(h, m)

 

모범답안

def solution(n,t,m,timetable):
    timetable=[int(i[:2])*60+int(i[3:]) for i in timetable]
    timetable.sort()
    bustable=[9*60+t*i for i in range(n)]
    for i in bustable:
        passenger=[p for p in timetable if p<=i]
        if i==bustable[-1]:
            if len(passenger)>=m:
                answer=passenger[m-1]-1
            elif len(passenger)<m:
                answer=i
            else:
                answer=passenger[-1]
        elif len(passenger)<m:
            timetable=timetable[len(passenger):]
        elif len(passenger)>=m:
            timetable=timetable[m:]
    answer= str(divmod(answer,60)[0]).rjust(2,'0')+':'+str(divmod(answer,60)[1]).rjust(2,'0')
    return answer

---

※ 알아야 할 것

-  다음번에 시간에 관한 구현은 '{0:02d}:{1:02d}'.format(h, m) 을 이용해보자!