알고리즘 연습 [셔틀 버스]
카카오 공채 1-4 셔틀 버스
문제
카카오에서는 무료 셔틀버스를 운행하기 때문에 판교역에서 편하게 사무실로 올 수 있다. 카카오의 직원은 서로를 ‘크루’라고 부르는데, 아침마다 많은 크루들이 이 셔틀을 이용하여 출근한다.
이 문제에서는 편의를 위해 셔틀은 다음과 같은 규칙으로 운행한다고 가정하자.
셔틀은 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 사이의 값이 될 수 있다.
나의 풀이
def get_bus_time_table(n, t):
h = 9
m = 0
result = [[9, 0]]
for i in range(n - 1):
m += t
if m > 60:
h += m // 60
m = m % 60
result.append([h, m])
return result
def get_str_time(t):
# 분 처리
if t[1] < 0:
t[1] = 60 + t[1]
t[0] -= 1
if t[1] >= 60:
t[1] = t[1] - 60
t[0] += 1
h = '0' + str(t[0]) if len(str(t[0])) == 1 else str(t[0])
m = '0' + str(t[1]) if len(str(t[1])) == 1 else str(t[1])
return h + ':' + m
def question_4(n, t, m, table):
# n:운행횟수 t:간격 m:사람수
crews = []
for s in table:
s = s.split(':')
crews.append(list(map(lambda i: int(i), s)))
bus_t = get_bus_time_table(n, t)
# 시간 순으로 정렬
crews = sorted(crews, key=lambda x: x[1])
crews = sorted(crews, key=lambda x: x[0])
p = []
for i in range(n):
p.append(m)
for i, t in enumerate(bus_t):
for j, c in enumerate(crews):
if c[0] > 0 and t[0] > c[0]:
# 버스시간보다 일찍 온 크루순으로 보냄
crews[j] = [-1]
p[i] -= 1
elif t[0] == c[0] and t[1] >= c[1]:
crews[j] = [-1]
p[i] -= 1
elif bus_t[-1][0] < c[0]:
# 버스 마지막 시간보다 늦게 온 크루도 없앰
crews[j] = [-1]
elif t[0] == c[0] and t[1] >= c[1]:
crews[j] = [-1]
if p[-1] == 1:
# 마지막 시간의 한 자리가 남았을 때
if crews[:] != [[-1] for i in range(len(crews))]:
# 타지 못한 크루가 남아있으면 -> 마지막 한 자리를 차지할 크루보다 1분 빨리 와야함
temp = [crews[j + 1][0], crews[j + 1][1] - 1]
return get_str_time(temp)
if p[i] == 0:
break
# 마지막 버스의 자리가 남아있으므로 마지막 버스시간에 탑승
return get_str_time(bus_t[-1])