[Programmers/파이썬]프로그래머스(Lv.2) 주차요금계산 문제풀이
문제 설명
주차장의 요금표와 차량이 들어오고(입차) 나간(출차) 기록이 주어졌을 때, 차량별로 주차 요금을 계산하려고 합니다. 아래는 하나의 예시를 나타냅니다.
- 요금표
기본 시간(분)기본 요금(원)단위 시간(분)단위 요금(원)
| 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)에 입차되는 경우는 입력으로 주어지지 않습니다.
- 아래의 예를 포함하여, 잘못된 입력은 주어지지 않습니다.
- 주차장에 없는 차량이 출차되는 경우
- 주차장에 이미 있는 차량(차량번호가 같은 차량)이 다시 입차되는 경우
입출력 예
| [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] |
사고 흐름
차량번호: 누적시간 자료구조를 가진 딕셔너리를 만들어야겠다
-> 누적 시간을 어떻게 구할 것 인가?
-> 동일한 차 번호를 가진 문자열에서 시간을 추출해 계산을 하자
-> 처음에는 해당 차 번호에 대해 모든 시간을 추가해 다시 한번 계산하는 과정을 만들었지만 비효율적이라 판단하고 바로 for문을 돌며 바로 계산하는 걸로 바꿈 (나는 hour*60+min 으로 했는데 다른사람들은?)
-> 차량번호로 정렬하고 요금 계산 진행
문제 핵심과 알고리즘
특정 알고리즘 문제라기 보단 모든 리스트를 돌며 효율적으로 원하는 정보를 추출하느냐의 문제이고, 문자열과 리스트 활용 문제같다.
import math
def solution(fees, records):
answer = []
cars=set()
for (i, r) in enumerate(records) : #차 번호 추출
cars.add(r[6:10])
car_dict={} #차번호: 누적시간
for (i, c) in enumerate(cars) :
cnt=0
car_dict[c]=0
for (j, r) in enumerate(records) :
if r[6:10]==c : #해당 차 번호를 가진 문자열에서 시간 계산
if cnt%2==0 : #처음엔 무조건 in
car_dict[c]-=int(r[0:2])*60+int(r[3:5])
cnt+=1
else : #홀수는 out
car_dict[c]+=int(r[0:2])*60+int(r[3:5])
cnt+=1
if cnt%2!=0 : # 만약 홀수로 끝났다면 out이 없으므로 23:59 계산 추가
cars_time[i]+=1439
#차량번호로 정렬
car_dict={k:v for k,v in sorted(car_dict.items(), key=lambda x:x[0])}
for (k, v) in car_dict.items() : #딕셔너리 돌면서 요금계산
sum=fees[1]
if (v-fees[0])>0 :
sum+=math.ceil((v-fees[0])/fees[2])*fees[3]
answer.append(sum)
return answer문제 풀면서 헤맸던 부분
1. 올림
요금 계산을 할 때 올림을 해야하는데, 처음에는 int()+1로 했다. 근데 정수일 때를 생각 못했고 자꾸 틀렸다. 그래서 구글링으로 ceil() 함수를 찾았고 이를 사용했다. 암기하자..!
다른 사람 풀이 참고하며 배운 점
import math
def solution(fees, records):
check = {}
for record in records:
time, number, status = record.split()
time = time.split(':')
time = int(time[0])*60 + int(time[1])
if number not in check:
check[number] = (0, time, status)
if status == 'IN':
check[number] = (check[number][0], time, status)
elif status == 'OUT':
total_time, in_time, _ = check[number]
total_time += time - in_time
check[number] = (total_time, time, status)
result = {}
for number in check.keys():
total_time, time, status = check[number]
if status == 'IN':
total_time += 1439 - time
fee = fees[1]
if total_time <= fees[0]:
result[number] = fee
else:
fee = fee + math.ceil((total_time - fees[0]) / fees[2]) * fees[-1]
result[number] = fee
return list(map(lambda x : x[1], sorted(result.items())))나는 먼저 차량번호를 추출하고 해당 차량번호에 맞는 시간들을 계산했지만 이러면 이중for을 써서 비효율적으로 보인다. 이 풀이는 records를 돌면서 시간, 차량번호, 내역을 가져오고 처음 차량이면 check를 하고 입차와 출차일때 나눠 시간계산을 진행했다.
총평
문제가 조금 길어서 차근차근 정리하면서 읽자. 그리고 단계를 나눠 코드를 짜자
<개념 정리>
1. 올림, 반올림, 내림, 소수점 버림
import math
#올림
math.ceil(3.14) #4
math.ceil(-3.14) $-3
#반올림
round(3.14) #3
round(3.6789, 2) #3.68
#내림
math.floor(3.14) #3
math.floor(-3.14) #-4
#소수점 버림
math.trunc(12.34) #12
math.trunc(-12.34) #-12