[파이썬 python] 프로그래머스 - 셔틀버스

• [1차] 셔틀버스

문제 설명

셔틀버스

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

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

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

입출력 예제

ntmtimetableanswer

1	1	5	["08:00", "08:01", "08:02", "08:03"]	"09:00"
2	10	2	["09:10", "09:09", "08:00"]	"09:09"
2	1	2	["09:00", "09:00", "09:00", "09:00"]	"08:59"
1	1	5	["00:01", "00:01", "00:01", "00:01", "00:01"]	"00:00"
1	1	1	["23:59"]	"09:00"
10	60	45	["23:59","23:59", "23:59", "23:59", "23:59", "23:59", "23:59", "23:59", "23:59", "23:59", "23:59", "23:59", "23:59", "23:59", "23:59", "23:59"]	"18:00"

해설 보러가기

카카오 신입 공채 1차 코딩 테스트 문제 해설

---

def solution(n, t, m, timetable):
    answer = ''
    # timetable을 int형 분 단위로 전환하였음
    new_timetable = []
    for i, time in enumerate(timetable):
        new_timetable.append( int(time[0:2])*60 + int(time[3:]) )
    new_timetable.sort(reverse = True)
    
    # bus 시간 테이블을 만들었음(int형 분 단위로)
    bus_table = [540+(i*t) for i in range(n)]
    
    tmp_answer = 0
    bus_idx = 0
    # 처음 버스 부터 마지막 버스 까지 움직인다.
    while bus_idx < len(bus_table):
    	# 현재의 버스라고 가정함.
        stack = []
        bus_onboard = 0
        # 해당하는 버스에 정원이 다 찰때까지 인원을 세줌 
        while bus_onboard != m:
        	# 조건에 맞다면 인원을 버스에 태워준다.
            if len(new_timetable) >= 1 and new_timetable[len(new_timetable)-1] <= bus_table[bus_idx]:
                bus_onboard += 1
                # 버스에 탄다
                stack.append(new_timetable.pop())
            else:
                break
        # 만약 마지막 버스인데,
        if bus_idx == len(bus_table)-1:
        	# 정원이 다 찼다면
            if bus_onboard == m:
            	# 마지막으로 탄 한 명을 잡아댕겨서 빼주고 탈 준비.
                # 나는 무조건 같은 시간대 마지막으로 타기 때문에, 마지막 사람 - 1 을 해줘야함.
                tmp_answer = stack.pop() - 1
            # 정원이 다 안찼다면
            else:
            	# 천천히 갈 수 있으므로 최대한 늦게(버스 시간에 맞춰서)
                tmp_answer = bus_table[bus_idx]
        
        bus_idx += 1
    
    # int형 분단위를 string형 답으로 변환함.
    if tmp_answer//60 < 10:
        answer += '0'+str(tmp_answer//60)
    else:
        answer += str(tmp_answer//60)
    answer += ":"
    if tmp_answer%60 < 10:
        answer += '0'+str(tmp_answer%60)
    else:
        answer += str(tmp_answer%60)
    
    #print(new_timetable)
    #print(bus_table)
    #print(stack)
    return answer

 

정확성  테스트
테스트 1 〉	통과 (0.04ms, 10.5MB)
테스트 2 〉	통과 (0.04ms, 10.4MB)
테스트 3 〉	통과 (0.04ms, 10.6MB)
테스트 4 〉	통과 (0.04ms, 10.5MB)
테스트 5 〉	통과 (0.06ms, 10.4MB)
테스트 6 〉	통과 (0.04ms, 10.5MB)
테스트 7 〉	통과 (0.67ms, 10.5MB)
테스트 8 〉	통과 (0.03ms, 10.4MB)
테스트 9 〉	통과 (0.03ms, 10.5MB)
테스트 10 〉	통과 (0.03ms, 10.4MB)
테스트 11 〉	통과 (0.05ms, 10.3MB)
테스트 12 〉	통과 (0.61ms, 10.4MB)
테스트 13 〉	통과 (0.56ms, 10.3MB)
테스트 14 〉	통과 (0.09ms, 10.4MB)
테스트 15 〉	통과 (0.15ms, 10.4MB)
테스트 16 〉	통과 (0.25ms, 10.5MB)
테스트 17 〉	통과 (0.61ms, 10.6MB)
테스트 18 〉	통과 (0.52ms, 10.4MB)
테스트 19 〉	통과 (0.57ms, 10.5MB)
테스트 20 〉	통과 (0.53ms, 10.4MB)
테스트 21 〉	통과 (1.42ms, 10.5MB)
테스트 22 〉	통과 (0.67ms, 10.5MB)
테스트 23 〉	통과 (0.51ms, 10.4MB)
테스트 24 〉	통과 (1.31ms, 10.5MB)

 

후기

 특정 규칙을 찾아야만 제대로 풀 수 있는 문제다.

 헛발질하고 삽질 한 후에 거의 2시간을 삽질했다. 처음에 그냥 간단해보였는데 점점 코드를 적으면서 뭔가 이상함을 느꼈고 점점 안드로메다로 향했다. 진짜로 코딩테스트였으면 시간이 부족해서 못 풀었을 듯 싶다.

 문제를 처음에 이해를 잘 못했고, 풀이 방법도 제대로 짜지 못한 채로 푸는 것은 삼가하자... 

 

 

 

풀이

* 딱 2가지 케이스로 정답을 생각해낼 수 있는 문제다. 이 규칙을 찾지 못하면 맞출 수가 없다.

* case1) 마지막 버스에 정원이 다 찼는가? -> 마지막으로 탄 한명을 잡아댕긴다. 

* case2) 마지막 버스에 정원이 다 안찼는가? -> 천천히 가준다. 버스시간에 맞춰준다.

 

(1) 정렬되어있는 인원들의 timetable을 준비한다. 또한 버스의 timetable 도 생성해준다.

(2) 처음 버스부터 마지막 버스까지, 시간별로 버스에 들어간다.

(3) 해당 시간별로 (시간에 맞게) 들어갈 수 있는 인원을 넣어주고 인원을 timetable에서 없애준다. 시간에 맞지 않거나, 시간별 버스 정원이 다 찼을 경우는, 다음 버스로 간다.  

(4) 마지막 버스일 경우, case1) 일 경우 (마지막으로 탄 한명의 시간 - 1) 이 정답이다. 나는 같은 시간대에는 무조건 늦게 타기 때문에, 무조건적으로 마지막 사람보다 -1분을 해야 한다.

(5) 마지막 버스일 경우, case2) 일 경우, 천천히 버스시간에 맞추어 타면 된다. 즉 버스시간이 답이다.