[1차] 셔틀버스
https://school.programmers.co.kr/learn/courses/30/lessons/17678
문제 설명
셔틀은 09:00부터 총 n회 t분 간격으로 역에 도착하며, 하나의 셔틀에는 최대 m명의 승객이 탈 수 있다.
셔틀은 도착했을 때 도착한 순간에 대기열에 선 크루까지 포함해서 대기 순서대로 태우고 바로 출발한다. 예를 들어 09:00에 도착한 셔틀은 자리가 있다면 09:00에 줄을 선 크루도 탈 수 있다.
콘이 셔틀을 타고 사무실로 갈 수 있는 도착 시각 중 제일 늦은 시각을 구하여라.
단, 콘은 게으르기 때문에 같은 시각에 도착한 크루 중 대기열에서 제일 뒤에 선다. 또한, 모든 크루는 잠을 자야 하므로 23:59에 집에 돌아간다. 따라서 어떤 크루도 다음날 셔틀을 타는 일은 없다.
입력 형식
0 ＜ n ≦ 10
0 ＜ t ≦ 60
0 ＜ m ≦ 45
timetable은 최소 길이 1이고 최대 길이 2000인 배열로, 하루 동안 크루가 대기열에 도착하는 시각이 HH:MM 형식으로 이루어져 있다.
크루의 도착 시각 HH:MM은 00:01에서 23:59 사이이다.
풀이 설명
- 사람들의 도착 시간과 버스 도착 시간을 분 단위로 계산합니다.
- 버스 시간마다 조건에 맞는 인원은 탑승시킵니다.
def solution(n, t, m, timetable):
timeT = sorted([int(i[:2])*60 + int(i[-2:]) for i in timetable])
busT = [540+t*i for i in range(n)]
for bus in busT:
# 탑승 가능 인원, 버스 도착 시간 같거나 빨리 도착하는 인원
p = [i for i in timeT if i <= bus]
# 마지막 버스인 경우
if bus == busT[-1]:
if len(p) >=m: # 마지막인데 탑승 인원이 더 많으면 컷
answer = p[m-1]-1
elif len(p) < m: # 무난하게 남으면 버스 시간 맞춰서 와도 됨
answer = bus
else :
answer = p[-1]
# 탑승 인원이 최대 인원보다 적을 때
elif len(p) < m :
timeT = timeT[len(p):]
elif len(p) >= m: # 다음 버스에 태워야 함
timeT = timeT[m:]
return f'{answer//60:02d}:{answer%60:02d}'