[프로그래머스] Lv2. 주차 요금 계산 파이썬

• 1. 2022 KAKAO BLIND RECRUITMENT
• 주차 요금 계산

문제 설명

문제 설명

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

• 요금표

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

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 함수를 완성해주세요.

 

내 풀이

import math
def solution(fees, records):
    answer = []
    default_t, default_fee, unit_t, unit_fee = fees[0], fees[1], fees[2], fees[3]
    records_split = []
    time_dict = {}
    result_dict = {}
    #입력받은 배열을 split하여 저장하기
    for record in records:
        records_split.append(record.split())
    #모든 차량 IN 처리. OUT이 있는 차량은 pop하고 시간 계산
    for record in records_split:
        status = record[2]
        if status == 'IN':
            time_dict[record[1]] = record[0]
        else:
            in_time = time_dict[record[1]]
            out_time = record[0]
            time_dict.pop(record[1])
            hours = int(out_time[:2]) - int(in_time[:2])
            minutes = int(out_time[3:]) - int(in_time[3:])
            time = hours * 60 + minutes
            if record[1] in result_dict:
                result_dict[record[1]] += time
            else:
                result_dict[record[1]] = time
    #마지막까지 dict에 남아있으면(OUT이 없는 차량) 23:59 출차
    if len(time_dict) != 0:
        remains = list(time_dict.items())
        for remain in remains:
            in_time = remain[1]
            hours = 23 - int(in_time[:2])
            minutes = 59 - int(in_time[3:])
            time = hours * 60 + minutes
            if remain[0] in result_dict:
                result_dict[remain[0]] += time
            else:
                result_dict[remain[0]] = time
    #차량 번호로 정렬 후 요금 계산
    results = sorted(result_dict.items())
    for result in results:
        fee = default_fee
        if result[1] > default_t:
            fee += math.ceil((result[1] - default_t) / unit_t) * unit_fee
        answer.append(fee)
    return answer

park_time, car_id, status = record.split()을 이용하여  첫번째 반복문과 두번째 반복문을 합칠 수 있다.

또는 각 반복문을 함수로 만들어 깔끔하게 만들 수 있다.