[COS PRO 1급] 2차 문제2) 지하철 기다리기

아현·2023년 8월 18일
0

Algorithm

목록 보기
377/400

연습문제

지하철 기다리기


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

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

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

□ 매개변수 설명

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

□ return 값 설명

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




# -*- coding: utf-8 -*-
# UTF-8 encoding when using korean
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(s)
		if subway_minute >= current_minute:
			answer = subway_minute - current_minute
			break
	if answer == INF:
		return -1
	return answer

profile
Studying Computer Science

0개의 댓글