algorithm_test/programmers_bus.py at main · alreadysons/algorithm_test · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
"""
문제 설명
셔틀버스
카카오에서는 무료 셔틀버스를 운행하기 때문에 판교역에서 편하게 사무실로 올 수 있다. 카카오의 직원은 서로를 '크루'라고 부르는데, 아침마다 많은 크루들이 이 셔틀을 이용하여 출근한다.

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

셔틀은 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 solution(n, t, m, timetable):
    answer = ''
    timetable.sort()
    timedetail = []
    for i in range(len(timetable)) :
        hours,minute = map(int,timetable[i].split(":"))
        timedetail.append(hours*60 + minute)
    if len(timetable) < m :
        if timetable[-1] < "09:00" :
            answer = "09:00"
        else :
            lasttime = n*t + 540 - t
            hour = str(lasttime // 60)
            minute = str(lasttime % 60)
            if len(hour) < 2 :
                hour = "0" + hour
            if len(minute) < 2 :
                minute = "0" + minute
            answer = hour + ":" + minute
    else :
        shuttle = [540 + i*t for i in range(n)]
        crew = []
        index = 0
        for s in shuttle :
            shuttlecrew = 0
            temp = []
            while index < len(timedetail) and timedetail[index] <= s and shuttlecrew < m :
                temp.append(timedetail[index])
                index +=1
                shuttlecrew += 1
            crew.append(temp)
        if len(crew[-1]) < m :
            lasttime = shuttle[-1]
            hour = str(lasttime // 60)
            minute = str(lasttime % 60)
            if len(hour) < 2 :
                hour = "0" + hour
            if len(minute) < 2 :
                minute = "0" + minute
            answer = hour + ":" + minute
        else :
            lasttime = crew[-1][-1] - 1
            hour = str(lasttime // 60)
            minute = str(lasttime % 60)
            if len(hour) < 2 :
                hour = "0" + hour
            if len(minute) < 2 :
                minute = "0" + minute
            answer = hour + ":" + minute
    return answer