프로그래머스 주차 요금 계산

문제 설명

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

• 요금표

기본 시간(분)기본 요금(원)단위 시간(분)단위 요금(원)

180

5000

10

600

 

• 입/출차 기록

시각(시:분)차량 번호내역

05:34	5961	입차
06:00	0000	입차
06:34	0000	출차
07:59	5961	출차
07:59	0148	입차
18:59	0000	입차
19:09	0148	출차
22:59	5961	입차
23:00	5961	출차

 

• 자동차별 주차 요금

차량 번호누적 주차 시간(분)주차 요금(원)

0000	34 + 300 = 334	5000 + ⌈(334 - 180) / 10⌉ x 600 = 14600
0148	670	5000 +⌈(670 - 180) / 10⌉x 600 = 34400
5961	145 + 1 = 146	5000

• 어떤 차량이 입차된 후에 출차된 내역이 없다면, 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)에 입차되는 경우는 입력으로 주어지지 않습니다.
  • 아래의 예를 포함하여, 잘못된 입력은 주어지지 않습니다.
    • 주차장에 없는 차량이 출차되는 경우
    • 주차장에 이미 있는 차량(차량번호가 같은 차량)이 다시 입차되는 경우

입출력 예

fees	records	result
[180, 5000, 10, 600]	["05:34 5961 IN", "06:00 0000 IN", "06:34 0000 OUT", "07:59 5961 OUT", "07:59 0148 IN", "18:59 0000 IN", "19:09 0148 OUT", "22:59 5961 IN", "23:00 5961 OUT"]	[14600, 34400, 5000]
[120, 0, 60, 591]	["16:00 3961 IN","16:00 0202 IN","18:00 3961 OUT","18:00 0202 OUT","23:58 3961 IN"]	[0, 591]
[1, 461, 1, 10]	["00:00 1234 IN"]	[14841]

입출력 예 설명

입출력 예 #1

문제 예시와 같습니다.

입출력 예 #2

• 요금표

기본 시간(분)기본 요금(원)단위 시간(분)단위 요금(원)

120

0

60

591

• 입/출차 기록

시각(시:분)차량 번호내역

16:00	3961	입차
16:00	0202	입차
18:00	3961	출차
18:00	0202	출차
23:58	3961	입차

• 자동차별 주차 요금

차량 번호누적 주차 시간(분)주차 요금(원)

0202	120	0
3961	120 + 1 = 121	0 +⌈(121 - 120) / 60⌉x 591 = 591

• 3961번 차량은 2번째 입차된 후에는 출차된 내역이 없으므로, 23:59에 출차되었다고 간주합니다.

입출력 예 #3

• 요금표

기본 시간(분)기본 요금(원)단위 시간(분)단위 요금(원)

1

461

1

10

• 입/출차 기록

시각(시:분)차량 번호내역

00:00

1234

입차

• 자동차별 주차 요금

차량 번호누적 주차 시간(분)주차 요금(원)

1234

1439

461 +⌈(1439 - 1) / 1⌉x 10 = 14841

• 1234번 차량은 출차 내역이 없으므로, 23:59에 출차되었다고 간주합니다.

풀이

카카오 문제 답지 않게 친절한 문제 ㅋㅋㅋ

import math

def solution(fees, records):
    answer = []    
    parking_in_out = {}
    parking_record = []
    
    for i in range(len(records)): # 시간 정리
        hour = int(records[i][:2])
        mins = int(records[i][3:5])
        carn = records[i][6:10]
        if carn in parking_in_out:
            parking_in_out[carn] += 1
        else:
            parking_in_out[carn] = 1
        parking_record.append([carn, (hour * 60 + mins)])
    
    for dic in parking_in_out.items(): # 출차 없는 애들 추가
        if dic[1] % 2 == 1:
            parking_record.append([dic[0], 23 * 60 + 59]) 
            parking_in_out[dic[0]] += 1
    
    parking_record.sort(key=lambda x:(x[0],x[1])) # 차량 번호 순 정렬
    
    i = 0
    while i < len(parking_record): # 요즘 정산
        fee, time = 0, 0
        for ii in range(0,parking_in_out[parking_record[i][0]],2):
            time += parking_record[i+ii+1][1]-parking_record[i+ii][1]
        if time <= fees[0]:
            fee = fees[1]
        else:
            time -= fees[0]
            fee = fees[1] + math.ceil(time / fees[2]) * fees[3]
        answer.append(fee)
        i += parking_in_out[parking_record[i][0]]
        
    return answer

문제는 어렵지 않았는데 전처리가 오래 걸렸다.

정확성  테스트
테스트 1 〉	통과 (0.06ms, 10.3MB)
테스트 2 〉	통과 (0.03ms, 10.4MB)
테스트 3 〉	통과 (0.08ms, 10.4MB)
테스트 4 〉	통과 (0.09ms, 10.3MB)
테스트 5 〉	통과 (0.25ms, 10.4MB)
테스트 6 〉	통과 (0.29ms, 10.4MB)
테스트 7 〉	통과 (2.20ms, 10.7MB)
테스트 8 〉	통과 (1.47ms, 10.6MB)
테스트 9 〉	통과 (0.27ms, 10.4MB)
테스트 10 〉	통과 (1.89ms, 10.7MB)
테스트 11 〉	통과 (2.76ms, 10.7MB)
테스트 12 〉	통과 (2.96ms, 10.7MB)
테스트 13 〉	통과 (0.04ms, 10.4MB)
테스트 14 〉	통과 (0.04ms, 10.3MB)
테스트 15 〉	통과 (0.03ms, 10.4MB)
테스트 16 〉	통과 (0.02ms, 10.4MB)

• 코드 스타일이 먼지다고 추천을 많이 받은 코드

from collections import defaultdict
from math import ceil

class Parking:
    def __init__(self, fees):
        self.fees = fees
        self.in_flag = False
        self.in_time = 0
        self.total = 0

    def update(self, t, inout):
        self.in_flag = True if inout=='IN' else False
        if self.in_flag:  self.in_time = str2int(t)
        else:             self.total  += (str2int(t)-self.in_time)

    def calc_fee(self):
        if self.in_flag: self.update('23:59', 'out')
        add_t = self.total - self.fees[0]
        return self.fees[1] + ceil(add_t/self.fees[2]) * self.fees[3] if add_t >= 0 else self.fees[1]

def str2int(string):
    return int(string[:2])*60 + int(string[3:])

def solution(fees, records):
    recordsDict = defaultdict(lambda:Parking(fees))
    for rcd in records:
        t, car, inout = rcd.split()
        recordsDict[car].update(t, inout)
    return [v.calc_fee() for k, v in sorted(recordsDict.items())]

• 다른 코드 역시 대부분 비슷하게 처리, 딱히 어려움 없었던 문제