[프로그래머스]주차 요금 계산 (python풀이)

이번엔 주차 요금 계산 문제 풀어보았습니다.

 

레벨 2 문제였으며 문제 푸는데 소요시간은 1시간 정도 걸렸네요.

 

문제 자체는 어렵지 않았으나 파이썬으로 문자열 다루는 게 어색해서 오래 걸린 거 같습니다. 

 

오늘부터는 문제 타입도 적어볼까합니다.

 

문제 타입: 

#시간 계산 #문자열 자르기

 

 

문제 설명

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

 

요금표

 

기본 시간(분)	기본 요금(원)	단위 시간(분)	단위 요금(원)
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

records 는 최대 1000개 가 입력되며 시각(시:분), 차량 번호, 내역만 공백으로 구분되어 입력됩니다. 

• 시각: HH:MM 형식 
• 차량번호: 4자리 숫자 형식
• 내역: IN / OUT

시간 기준으로 정렬되며 다음날 출차되는 경우는 없고 입차 된 후 출차 이력이 없다면 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]

 

해당 문제는 문제를 이해하는 건 어렵지 않습니다.

 

차량별로 입출차 기록을 확인하여 이용시간의 총합을 계산한 후

사용시간을 출력해주면 됩니다. 

차량별로 여러 번 이용한 케이스가 있어 해당 경우도 고려하였습니다.

 

이때 차량번호순으로 출력되어야 합니다. 

 

저의 경우 차량별 이용시간을 딕셔너리로 저장하였고

만약 출차 이력이 없는 경우 즉 이용시간의 개수가 홀수인 경우 마지막에 23:59를 추가로 저장해주었습니다. 

저의 경우 내역 칼럼을 고려하지 않았는데 해당 문제는 입 -> 출 -> 입... 이 반복되므로 기록의 개수로 입/출 판단을 하였습니다.

 

딕셔너리를 key로 접근하기 위해 차량번호를 sort 한 배열을 추가로 만들어주었으며 

차량번호별로 탐색해서 소요시간을 계산하였습니다. 

 

마지막에 합산 금액을 계산할 때 횟수별 별도 계산이 아니라 총 소요시간에서 금액을 산정해야 합니다. 

 

from collections import defaultdict
import math

def solution(fees, records):
    answer = []
    carNum = [] # 차량 번호 리스트
    carTime = defaultdict(list) # 차량별 시간. 리스트 형으로 초기화
    
    for i in range(0, len(records)):
        carNum.append(records[i][6:10])
        carTime[records[i][6:10]] += [int(records[i][0:2])*60+int(records[i][3:5])]
    
    carNum = list(set(carNum))
    carNum.sort()
    
    for i in range(0, len(carTime)): # 순서대로 출력해봄
        if len(carTime.get(carNum[i])) % 2 != 0:
            carTime.get(carNum[i]).append('1439') # 출차시간 없으면 넣어줌
        sum = 0
        for j in range(0, len(carTime.get(carNum[i]))-1,2): 
            diff = int(carTime.get(carNum[i])[j+1])-int(carTime.get(carNum[i])[j])

            sum += diff 
        
        if sum <= fees[0] :
            answer.append(fees[1])
        else :
            answer.append(fees[1] + math.ceil((sum-fees[0])/fees[2]) *fees[3])
        
    return answer