시각은 차량이 입차되거나 출차된 시각을 나타내며, HH:MM 형식의 길이 5인 문자열입니다.
HH:MM은 00:00부터 23:59까지 주어집니다.
잘못된 시각("25:22", "09:65" 등)은 입력으로 주어지지 않습니다.
차량번호는 자동차를 구분하기 위한, `0'~'9'로 구성된 길이 4인 문자열입니다.
내역은 길이 2 또는 3인 문자열로, IN 또는 OUT입니다. IN은 입차를, OUT은 출차를 의미합니다.
records의 원소들은 시각을 기준으로 오름차순으로 정렬되어 주어집니다.
records는 하루 동안의 입/출차된 기록만 담고 있으며, 입차된 차량이 다음날 출차되는 경우는 입력으로 주어지지 않습니다.
같은 시각에, 같은 차량번호의 내역이 2번 이상 나타내지 않습니다.
마지막 시각(23:59)에 입차되는 경우는 입력으로 주어지지 않습니다.
아래의 예를 포함하여, 잘못된 입력은 주어지지 않습니다.
주차장에 없는 차량이 출차되는 경우
주차장에 이미 있는 차량(차량번호가 같은 차량)이 다시 입차되는 경우
입출력 예
입출력 예 설명
입출력 예 #1 문제 예시와 같습니다.
입출력 예 #2
요금표
입/출차 기록
자동차별 주차 요금
3961번 차량은 2번째 입차된 후에는 출차된 내역이 없으므로, 23:59에 출차되었다고 간주합니다.
입출력 예 #3
요금표
입/출차 기록
자동차별 주차 요금
1234번 차량은 출차 내역이 없으므로, 23:59에 출차되었다고 간주합니다.
풀이
import collections
import math
def solution(fees, records):
answer = []
total_time = collections.defaultdict(int) #차량번호 별 이용시간 저장할 dict
check = collections.defaultdict(str) #차량 in,out 확인할 check
#기본시간, 기본요금, 단위시간, 단위요금
basic_t = int(fees[0])
basic_f = int(fees[1])
unit_t = int(fees[2])
unit_f = int(fees[3])
for record in records:
t, num, io = record.split() #i,o시간, 차량번호, i or o
if num not in check.keys(): #check 리스트에 차량번호가 없다면 -> in
check[num] = t
else: #차량번호가 있다면 out시간-in시간,
out_t = int(t.split(':')[0]) * 60 + int(t.split(':')[1])
in_t = int(check[num].split(':')[0]) * 60 + int(check[num].split(':')[1])
total_time[num] += out_t - in_t
del check[num] #out time 계산 끝난 후, check[num] 지우기
hours = int(23*60 + 59) #00:00 ~ 23:59까지 분으로 쳤을 때
#in만 있고 out이 없을 때
#check dict에 존재한다면
if check:
for i in check.keys():
in_t = int(check[i].split(':')[0]) * 60 + int(check[i].split(':')[1])
total_time[i] += hours - in_t
#차량번호 오름차순 순으로 가격을 계산하기 위해 total_time => lst로 변경
result = []
for i in total_time:
result.append([i,total_time[i]])
result.sort(key=lambda x:x[0])
#answer에 요금 넣기
for i in result:
temp_num = total_time[i[0]] #각 차량의 total_time
if temp_num > basic_t :
answer.append(basic_f + math.ceil((temp_num-basic_t)/unit_t)* unit_f)
else:
answer.append(basic_f)
return answer
풀이 설명
고려해야할 사항
차량이 입차만 하고 출차하지 않은 경우
차량 번호 오름차순으로 내야할 요금 return
생각 과정
1. 차량번호 별 이용시간을 저장할 dict와 차량의 in,out을 확인할 dict를 생성
2. 차량의 in,out이 기록된다면 out time - in time을 차량번호 별 이용시간 dict에 저장하기. 저장 후 차량 in,out 확인한 dict에서 해당 차량의 정보 삭제(이후 다시 입차할 시, in-out time 계산을 추가로 진행해야하기 때문)
2-1. 만약 in만 존재 시(out time이 없을 시), 23:59 - in time를 차량번호 별 이용시간 dict에 추가
3. 차량번호 오름차순으로 정렬하기 위해, 차량번호 별 이용시간 dict -> list로 변경, 차량번호 별 오름차순 정렬
