문제 설명
주차장의 요금표와 차량이 들어오고(입차) 나간(출차) 기록이 주어졌을 때, 차량별로 주차 요금을 계산하려고 합니다. 아래는 하나의 예시를 나타냅니다.
- 요금표
기본 시간(분) 기본 요금(원) 단위 시간(분) 단위 요금(원)
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 = 148411234번 차량은 출차 내역이 없으므로, 23:59에 출차되었다고 간주합니다.
제한시간 안내
정확성 테스트 : 10초
접근 방법
주차된 차량, 차량의 기록, 정산된 요금, 총 주차 시간을 collections의 defaultdict()를 사용하여 관리하였다. key는 차량 번호가 된다. defaultdict()를 사용함으로써 key error를 방지하였다.
records를 순회하며 값을 빈칸을 기준으로 split하고, 이 값들 중에서 시간을 나타내는 문자열을 ':'을 기준으로 split하여 차량 기록 딕셔너리에 [타입(IN/OUT), 시, 분]을 넣어준다. 이때 튜플이 아닌 리스트로 넣어줘야 다음에 있을 값 변경이 가능하다.
차량의 기록 딕셔너리를 순회하며 타입이 IN일 경우 주차된 차량에 [시, 분]을 넣어주고, OUT일 경우 주차된 차량에 저장된 [시, 분]과 현재의 시, 분을 계산하여 총 주차 시간 딕셔너리의 값을 증가시켜주고, 주차된 차량 딕셔너리에서 해당 차량을 삭제해준다.
위의 과정이 끝났을 때, 주차된 차량이 남아있을 경우, 주차된 차량 딕셔너리를 순회하며 [시, 분]을 23:59와 계산하여 총 주차 시간 딕셔너리를 증가시킨다.
마지막으로 총 주차 시간 딕셔너리를 순회하며 정산된 요금 딕셔너리를 계산하여 넣어주고, 정산된 요금 딕셔너리를 key의 오름차순으로 정렬한 후에 해당 value를 answer에 넣어주면 문제가 해결된다.
- 정답을 저장할 리스트 answer를 선언한다.
- 주차된 차량에 대한 딕셔너리 parked를
defaultdict(list)로 선언한다. - 차량의 정보에 대한 딕셔너리 in_out을
defaultdict(list)로 선언한다. - 정산한 요금에 대한 딕셔너리 result를
defaultdict(int)로 선언한다. - 총 주차 시간에 대한 딕셔너리 time을
defaultdict(int)로 선언한다. - records의 길이만큼 반복하는 i에 대한 for문을 돌린다.
-> 임시 변수 tmp에records[i]를 빈칸을 기준으로 split한 리스트를 저장한다.
-> h, m에tmp[0]을 ':'을 기준으로 split한 값을 int형으로 저장한다.
->in_out[tmp[1]]에[tmp[2], h, m]을 넣는다. - in_out을 순회하는 key, value에 대한 for문을 돌린다.
-> 만약value[i][0]이 IN일 경우,
-->parked[key]에[value[i][1], value[i][2]]를 넣는다.
-> 만약value[i][0]이 OUT일 경우,
--> 만약parked[key][0][1]이value[i][2]보다 클 경우,
--->value[i][1]을 1 감소시킨다.
--->value[i][2]를 60 증가시킨다.
--->time[key]를value[i][2]-parked[key][0][1]만큼 증가시킨다.
--->time[key]를(value[i][1]-parked[key][0][0])*60만큼 증가시킨다.
--> 그 외의 경우,
--->time[key]를value[i][2]-parked[key][0][1]만큼 증가시킨다.
--->time[key]를(value[i][1]-parked[key][0][0])*60만큼 증가시킨다.
--->parked[key]를 삭제한다. - 만약 parked가 존재할 경우,
-> parked를 순회하는 key, value에 대한 for문을 돌린다.
-->time[key]를59-value[0][1]만큼 증가시킨다.
-->time[key]를(23-value[0][0])*60만큼 증가시킨다. - time을 순회하는 key, value에 대한 for문을 돌린다.
-> 만약 value가fees[0]보다 작을 경우,
-->result[key]에fees[1]을 더한다.
-> 그 외의 경우,
--> 만약value-fees[0]을fees[2]로 나눈 나머지가 0보다 클 경우,
--->result[key]를((value-fees[0])//fees[2]+1)*fees[3]+fees[1]만큼 증가시킨다. (올림)
--> 그 외의 경우,
--->result[key]를((value-fees[0])//fees[2])*fees[3]+fees[1]만큼 증가시킨다. - result를 key의 오름차순으로 정렬한다.
- result를 순회하는 num, cost에 대한 for문을 돌린다.
-> answer에 cost를 넣는다. - answer를 반환한다.
solution.py
import collections
def solution(fees, records):
answer=[]
parked=collections.defaultdict(list)
in_out=collections.defaultdict(list)
result=collections.defaultdict(int)
time=collections.defaultdict(int)
for i in range(len(records)):
tmp=list(records[i].split())
h, m=map(int, tmp[0].split(':'))
in_out[tmp[1]].append([tmp[2], h, m])
for key, value in in_out.items():
for i in range(len(value)):
if value[i][0]=='IN':
parked[key].append([value[i][1], value[i][2]])
if value[i][0]=='OUT':
if parked[key][0][1]>value[i][2]:
value[i][1]-=1
value[i][2]+=60
time[key]+=value[i][2]-parked[key][0][1]
time[key]+=(value[i][1]-parked[key][0][0])*60
else:
time[key]+=value[i][2]-parked[key][0][1]
time[key]+=(value[i][1]-parked[key][0][0])*60
del parked[key]
if parked:
for key, value in parked.items():
time[key]+=59-value[0][1]
time[key]+=(23-value[0][0])*60
for key, value in time.items():
if value<=fees[0]:
result[key]+=fees[1]
else:
if (value-fees[0])%fees[2]>0:
result[key]+=((value-fees[0])//fees[2]+1)*fees[3]+fees[1]
else:
result[key]+=((value-fees[0])//fees[2])*fees[3]+fees[1]
result=sorted(result.items())
for num, cost in result:
answer.append(cost)
return answer
꾸준함을 꿈꾸는 SW 전공 학부생의 개발 일기