[programmers] 주차 요금 계산, 92341

문제 설명

주차장의 요금표와 차량이 들어오고(입차) 나간(출차) 기록이 주어졌을 때, 차량별로 주차 요금을 계산하려고 합니다. 아래는 하나의 예시를 나타냅니다.

 

- 요금표

- 입/출차 기록
시각(시:분) 차량 번호 내역

- 자동차별 주차 요금

• 어떤 차량이 입차된 후에 출차된 내역이 없다면, 23:59에 출차된 것으로 간주합니다.
  • 0000번 차량은 18:59에 입차된 이후, 출차된 내역이 없습니다. 따라서, 23:59에 출차된 것으로 간주합니다.
• 00:00부터 23:59까지의 입/출차 내역을 바탕으로 차량별 누적 주차 시간을 계산하여 요금을 일괄로 정산합니다.
• 누적 주차 시간이 기본 시간이하라면, 기본 요금을 청구합니다.
• 누적 주차 시간이 기본 시간을 초과하면, 기본 요금에 더해서, 초과한 시간에 대해서 단위 시간 마다 단위 요금을 청구합니다.
  • 초과한 시간이 단위 시간으로 나누어 떨어지지 않으면, 올림합니다.
  • ⌈a⌉ : a보다 작지 않은 최소의 정수를 의미합니다. 즉, 올림을 의미합니다.

주차 요금을 나타내는 정수 배열 fees, 자동차의 입/출차 내역을 나타내는 문자열 배열 records가 매개변수로 주어집니다. 차량 번호가 작은 자동차부터 청구할 주차 요금을 차례대로 정수 배열에 담아서 return 하도록 solution 함수를 완성해주세요.

제한사항

• fees의 길이 = 4
  • fees[0] = 기본 시간(분)
  • 1 ≤ fees[0] ≤ 1,439
  • fees[1] = 기본 요금(원)
  • 0 ≤ fees[1] ≤ 100,000
  • fees[2] = 단위 시간(분)
  • 1 ≤ fees[2] ≤ 1,439
  • fees[3] = 단위 요금(원)
  • 1 ≤ fees[3] ≤ 10,000
• 1 ≤ records의 길이 ≤ 1,000
  • records의 각 원소는 "시각 차량번호 내역" 형식의 문자열입니다.
  • 시각, 차량번호, 내역은 하나의 공백으로 구분되어 있습니다.
  • 시각은 차량이 입차되거나 출차된 시각을 나타내며, 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로 변경, 차량번호 별 오름차순 정렬
• 4. answer에 요금 넣기