[COS Pro 1급, Python] 2차 2번 : 지하철 기다리기

문제 유형

 빈칸

난이도

 easy

Note 

 Nothing

 

Code

# -*- coding: utf-8 -*-
# UTF-8 encoding when using korean
# 2차 2번

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

subway_times1 = ["05:31", "11:59", "13:30", "23:32"]
current_time1 = "12:00"
ret1 = solution(subway_times1, current_time1)

print("solution 함수의 반환 값은", ret1, "입니다.")

subway_times2 = ["14:31", "15:31"]
current_time2 = "15:31"
ret2 = solution(subway_times2, current_time2)

print("solution 함수의 반환 값은", ret2, "입니다.")

 

※ 가끔 코드 중 print(~)가 있습니다. 정리 못한 점 죄송합니다.