[Algorithm] Programmers : 주차 요금 계산 by Python

[문제 바로가기] https://programmers.co.kr/learn/courses/30/lessons/92341

코딩테스트 연습 - 주차 요금 계산

 

 

📌 문제 설명

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

• 요금표

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

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

 

---

💡 문제 풀이

문제를 풀기 위해서 아래의 세 가지를 순서대로 구현했다.

1. 입/출차에 따른 주차요금 계산

2. 출차 기록이 없는 차량에 대한 주차요금 계산

3. 차량 번호가 작은 자동차부터 청구할 주차 요금을 차례대로 배열에 담아 return

 

먼저, 시간을 계산하는 문제가 나올 경우 하나의 단위로 통일시켜주는 것이 좋다.

ex) [HH시:MM분] 형태의 데이터가 주어질 때, 1시간 = 60분 이므로, HH시 * 60 + MM분 

split 함수를 이용하여 ":"을 기준으로 시, 분을 분리하였다.

주차장의 "IN", "OUT"에 따른 시간 계산은 calculate 함수로 구현하였다.

 

1. 입/출차에 따른 주차요금 계산

자동차의 입/출차 내역으로 주어지는 records 배열이 차량번호를 기준으로 주어지는 것이 아닌, 랜덤으로 주어지기 때문에 특정 차량번호의 주차요금을 계산하기 위해서는 차량번호:[최근 입/출차 시간, 입/출차 상태, 사용시간(분)] 형태의 딕셔너리를 사용하는 것이 효율적이라고 판단 → 'cars'라는 딕셔너리 생성 

ex) records배열의 첫 번째 데이터인 05:34 | 5961 | 입차와 같은 데이터가 주어졌을 경우 → 5961 : ["05:34", "IN", 0]의 형태로 데이터를 담음

 

2. 출차 기록이 없는 차량에 대한 주차요금 계산

출차 기록이 없는 차량을 찾기위해서는 마지막 입/출차 상태가 "IN"인 차량번호를 찾으면 되기 때문에

1번 과정을 마친 'cars' 딕셔너리를 반복문을 이용하여 배열의 두 번째 값이 "IN"인 데이터를 찾아 "23:59" 출차된 것으로 간주하여 시간을 계산한다.

 

1번 ~ 2번 과정을 통해 시간을 모두 계산했으면 주차요금을 계산한다.

만약, 주차장 사용시간이 기본 시간(분)보다 크거나 같다면, 기본요금 + (사용시간(분) - 기본시간(분)) x 단위 요금

만약, 주차장 사용시간이 기본 시간(분)보다 작다면, 기본요금

 

3. 차량 번호가 작은 자동차부터 청구할 주차 요금을 차례대로 배열에 담아 return

파이썬에서 배열 또는 딕셔너리 등을 정렬할 때 사용하는 sorted함수를 이용함.

차량 번호가 cars 딕셔너리의 key값이므로 key값을 기준으로 오름차순 정렬한 후 2번에서 계산한 주차요금을 차례대로 answer 배열에 담아서 return하면 끝 

 

코드는 다음과 같다.

from math import ceil

# 시간단위를 '분'으로 변경
def calculate(time):
    hour, minute = time.split(":")
    return int(hour) * 60 + int(minute)

def solution(fees, records):
    cars = dict()
    answer = []
    for record in records:
        time, num, flag = record.split()
        # 입차인 경우
        if flag == "IN":
            # 최초 입차인 경우 초기세팅
            if num not in cars.keys():
                cars[num] = [time, flag, 0]
            else:
                cars[num][0], cars[num][1] = time, flag
        # 출차인 경우
        else:
            cars[num][2] += calculate(time) - calculate(cars[num][0])
            cars[num][1] = "OUT"

    # 출차 기록이 없는 경우 → "23:59"이 출차인 것으로 간주
    for key, value in cars.items():
        if value[1] == "IN":
            cars[key][2] += calculate("23:59") - calculate(value[0])

    # 주차 요금 계산
    for key, value in cars.items():
        if value[2] >= fees[0]:
            cars[key][2] = fees[1] + ceil((value[2] - fees[0]) / fees[2]) * fees[3]
        else:
            cars[key][2] = fees[1]

    # 차량 번호가 작은 자동차부터 주차 요금 담기
    for num in sorted(cars.keys()):
        answer.append(cars[num][2])

    return answer