본문 바로가기
Python/알고리즘

[COS PRO 1급 #2-2] 지하철 기다리기

by 포푸리 (POPOOLY) 2023. 2. 16.
반응형

문제설명

A 지하철역의 오늘 하루 지하철 도착 시각이 순서대로 들어있는 리스트가 있습니다. 현재 시간이 주어졌을 때, 지하철을 타기위해서는 최소 몇 분을 기다려야 하는지 구하려 합니다. 이를 위해 다음과 같이 프로그램 구조를 작성했습니다.

~~~
1. 00:00을 기준으로 해서 현재 시각을 분 단위로 변환합니다.
2. 리스트를 순회하며 다음을 수행합니다.
  2-1. 00:00을 기준으로, 각 지하철 도착 시각을 분 단위로 변환합니다.
  2-2. 현재 시각과 지하철 도착 시각을 비교하여 최소 대기 시간을 구합니다.
3. 2번 단계에서 구한 최소 대기 시간을 return 합니다.
~~~

오늘 하루 동안의 지하철 도착 시각이 순서대로 들어있는 리스트 subway_times와 현재시간 current_time이 매개변수로 주어질 때, 지하철을 타기 위해 기다려야 하는 최소 대기 시간을 return 하도록 solution 함수를 작성했습니다. 이때, 위 구조를 참고하여 중복되는 부분은 func_a라는 함수로 작성했습니다. 코드가 올바르게 동작할 수 있도록 빈칸을 알맞게 채워 전체 코드를 완성해주세요.

매개변수 설명

오늘 하루 동안의 지하철 도착 시각이 순서대로 들어있는 리스트 subway_times와 현재시간 current_time이 solution 함수의 매개변수로 주어집니다. 
* subway_times의 길이는 1 이상 100 이하입니다.
* subway_times에는 지하철 도착 시간이 중복없이 들어있습니다.
* 지하철 도착 시각과 현재 시각은 24시간 표기법을 이용하여 나타냅니다.
  * 24시간 표기법은 00:00부터 23:59까지로 시간을 표기합니다.
  * 입력으로 주어지는 시간은 항상 "XX:YY"형태로 주어집니다.
* 지하철 첫차 도착 시각은 05:00 이후이며, 막차 도착 시각은 23:59 이전입니다.

return 값 설명

지하철을 타기 위해 기다려야 하는 최소 대기 시간을 return 해주세요.
* 만약 오늘 탈 수 있는 지하철이 없다면 -1을 return 하면 됩니다.

예시

| subway_times                | current_time | return |
|-----------------------------|--------------|--------|
| ["05:31", "11:59", "13:30", "23:32"] | "12:00"      | 90     |
| ["14:31", "15:31"]          | "15:31"      | 0      |

예시 설명

예시 #1
현재시간 12:00에서 가장 가까운 지하철 도착 시각은 13:30입니다.
따라서 최소 90분을 기다려야 합니다.

예시 #2
현재 시각은 15:31이며, 15:31분에 도착하는 지하철에 탑승할 수 있습니다. 따라서 0분만 기다리면 됩니다.

문제 코드

def func_a(times):
    hour = int(times[:2])
    minute = int(times[3:])
    return hour*60 + minute

def solution(subway_times, current_time):
    current_minute = func_a(@@@)
    INF = 1000000000
    answer = INF
    for s in subway_times:
        subway_minute = func_a(@@@)
        if @@@:
            answer = subway_minute - current_minute
            break
    if answer == INF:
        return -1
    return answer

풀이

# 00:00을 기준으로 현재 시각을 분 단위로 변환
def func_a(times):
    hour = int(times[:2])
    minute = int(times[3:])
    return hour*60 + minute

def solution(subway_times, current_time):
    current_minute = func_a(current_time)
    INF = 1000000000
    answer = INF
    for s in subway_times:
        subway_minute = func_a(subway_times)
        # subway_minute이 current_minute보다 크면 그 차이가 대기 시간
        if subway_minute >= current_minute:
            answer = subway_minute - current_minute
            break
    if answer == INF:
        return -1
    return answer

댓글