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

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

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

2018 카카오 블라인드 채용 문제다.

[문제 설명]

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

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

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

 

[아이디어]

1) 마지막 한 자리를 남겨놓을 때까지 인원을 태우고 난 뒤의 timetable 상황에 따라 조건을 부여하자.

2) 도착한 시간의 역순으로 배열한 뒤 뒤에서 하나씩 빼내면 O(1)의 복잡도로 삭제를 할 수 있다.

3) 한 자리를 남겨놓고 배열의 맨 뒤에 위치한 사람이 버스 시간보다 일찍 왔다면 그 사람보다 1분 먼저 도착해야하고, 그렇지 않다면 그냥 버스 시간에 맞춰 도착하면 된다.

4) 사실 한 자리를 남겨놓고 사람을 태웠을 때 배열이 비어있는 경우를 고려해야 할 것 같은데, 해당 경우는 테스트케이스에 존재하지 않는 것 같다. (존재했다면 timetable[-1]에서 에러가 났을 것이다.)

 

# 정답
def solution(n, t, m, timetable):
    now = [9,0]
    timetable.sort(reverse = True)
    while n>1:
        n -= 1
        for _ in range(m):
            if (int(timetable[-1][:2])*60 +int(timetable[-1][-2:])) <= (now[0]*60 + now[1]):
                timetable.pop() 
        now[0], now[1] = now[0] + (now[1] + t)//60, (now[1] + t)%60
    for _ in range(m-1):
        if (int(timetable[-1][:2])*60 +int(timetable[-1][-2:])) <= (now[0]*60 + now[1]):
            timetable.pop() 
    if (len(timetable) > 0) and (int(timetable[-1][:2])*60 +int(timetable[-1][-2:])) <= (now[0]*60 + now[1]):
        answer = timetable[-1]
        if answer[-2:] == "00":
            answer = str(int(answer.split(":")[0]) -1).zfill(2) + ":59"
        else:
            answer = answer.split(":")[0].zfill(2) + ":" + str(int(answer.split(":")[1]) -1).zfill(2)
    else:
        answer = str(now[0]).zfill(2) + ":" + str(now[1]).zfill(2)
    return answer

생각보다 수월하게 해결했다. 다만 순서대로 정렬하는 것을 맨 위에 해놓고 함수에 포함시키지 않아 시간을 낭비했다. 형식에 맞게 str을 변형하는 zfill도 기억해두면 좋겠다.