[programmers/Lv.2] 주차 요금 계산

❓  문제

https://school.programmers.co.kr/learn/courses/30/lessons/92341

 

🛎️  아이디어

1. 출차 기록이 없는 차 번호를 찾는다.
    a. 주차장에 이미 있는 차량(차량번호가 같은 차량)이 다시 입차되는 경우가 없기 때문에 입출차 개수가 홀수라면 출차 기록이 한 개 부족한 것이다.
    b. 23:39로 출차 기록을 추가해준다.

2. 차량 번호 - 시간 - 내역 순으로 정렬을 한다.
    a. 내역 정렬을 하면 알파벳 순으로 IN - OUT이 정렬된다.

3. 차량 별 주차 누적 시간을 계산한다.
    a. 출차 시간인 경우 앞 인덱스의 시간(입차)의 차이를 구해서 더해준다.

4. 최종 요금을 구해준다.
    a. 올림을 사용한다고 했기 때문에 math를 임포트해서 ceil 함수를 사용한다.

 

💻  풀이

import math
from collections import Counter

def solution(fees, records):
  default_time, default_fee, unit_time, unit_fee = fees

  # 차 번호 별 입출차 갯수 계산
  car_numbers = dict(Counter(map(lambda x: x.split()[1], records)))

  # 출차 기록이 없는 번호는 23:59 추가
  for car in car_numbers.items():
    if car[1] % 2 != 0:
      records.append(f'23:59 {car[0]} OUT')

  # 차량 번호, 시간, 내역 순으로 정렬
  records.sort(key=lambda x: (x.split()[1], x.split()[0], x.split()[2]))
  # ['06:00 0000 IN', '06:34 0000 OUT', '18:59 0000 IN', '23:59 0000 OUT', '07:59 0148 IN', '19:09 0148 OUT', '05:34 5961 IN', '07:59 5961 OUT', '22:59 5961 IN', '23:00 5961 OUT']

  # 차량 별 누적 시간
  sum_time = {}
  for index, record in enumerate(records):
    time, car, history = record.split()

    # 출차 시간인 경우
    if history == "OUT":
      # 입차 시간
      in_time = records[index - 1].split()[0]
      sub_hour = int(time.split(':')[0]) - int(in_time.split(':')[0])
      sub_minute = int(time.split(':')[1]) - int(in_time.split(':')[1])

      # 실제 주차 시간
      sub_time = 60 * sub_hour + sub_minute

      if car in sum_time:
        sum_time[car] += sub_time
      else:
        sum_time[car] = sub_time

  answer = []
  for time in sum_time.values():
    # 최종 요금
    if time <= default_time:
      answer.append(default_fee)
    else:
      answer.append(default_fee + math.ceil((time - default_time) / unit_time)  * unit_fee)

  return answer

print(solution([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"]))