[프로그래머스/Python파이썬] 셔틀버스(2018 카카오 코딩테스트 1차)

문제 링크: programmers.co.kr/learn/courses/30/lessons/17678

코딩테스트 연습 - [1차] 셔틀버스

 

 

문제 설명

시간 계산을 쉽게 하기 위해서 "HH:MM" 형식의 시간을 모두 분으로 변환해서 생각했습니다. 

예를 들어 "09:10"인 경우 9*60 + 10 = 550(분)으로 변환합니다. 

이런 식으로 변환하면 정수 비교 연산을 통해 누가 먼저 왔는지를 판단할 수 있습니다. 

 

• crewTime 리스트에 각 크루가 도착하는 시간을 저장합니다. 
• 먼저 온 크루가 앞에 오도록 crewTime을 정렬합니다. 
• busTime 리스트에 9:00부터 t분 간격으로 n번 도착하는 버스 시간을 저장합니다. 
• 버스가 도착할 때마다(busTime의 요소마다) 버스에 타는 크루의 수(cnt)를 구합니다. 
• 버스에 자리가 남는 경우 콘은 버스 도착 시간에 도착하면 됩니다. 
• 버스에 자리가 없을 경우 콘은 맨 마지막에 버스에 탄 크루보다 1분 먼저 도착하면 됩니다. 

 

버스에 타는 크루의 수를 계산하는 코드는 다음과 같습니다. 

while cnt < m and i < len(crewTime) and crewTime[i] <= tm:
	i += 1
	cnt += 1

이때 각 조건은 다음을 의미합니다. 

cnt < m			# 버스에 자리가 남아있다
i < len(crewTime)	# 탑승할 크루가 남아있다
crewTime[i] <= tm	# 해당 크루가 버스 도착 시간 전에 도착했다

 

 

 

 

소스코드

def solution(n, t, m, timetable):
    answer = 0
    # 모든 시간을 분으로 환산해서 생각
    # 예시: "09:10" => 9*60 + 10 = 550(분)
    # 크루 도착 시각 리스트
    crewTime = [int(time[:2])*60 + int(time[3:]) for time in timetable]
    crewTime.sort()
    # 버스 도착 시각 리스트
    busTime = [9*60 + t*i for i in range(n)]

    i = 0       # 다음에 버스에 오를 크루의 인덱스
    for tm in busTime:
      cnt = 0   # 버스에 타는 크루 수
      while cnt < m and i < len(crewTime) and crewTime[i] <= tm:
        i += 1
        cnt += 1
      # 버스에 자리가 남았을 경우
      if cnt < m: answer = tm
      # 버스에 자리가 없는 경우 맨 마지막 크루보다 1분 먼저 도착
      else: answer = crewTime[i - 1] - 1

    return str(answer//60).zfill(2) + ":" + str(answer%60).zfill(2)