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

Donghyun·2024년 9월 5일
코드카타파이썬프로그래머스
0

Code Kata - 파이썬

목록 보기
51/54
post-thumbnail

링크: https://school.programmers.co.kr/learn/courses/30/lessons/92341?language=python3

문제 설명

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

  • 요금표
기본 시간(분)기본 요금(원)단위 시간(분)단위 요금(원)
180500010600
  • 입/출차 기록
시각(시:분)차량 번호내역
05:345961입차
06:000000입차
06:340000출차
07:595961출차
07:590148입차
18:590000입차
19:090148출차
22:595961입차
23:005961출차
  • 자동차별 주차 요금
차량 번호누적 주차 시간(분)주차 요금(원)
000034 + 300 = 3345000 + ⌈(334 - 180) / 10⌉ x 600 = 14600
01486705000 +⌈(670 - 180) / 10⌉x 600 = 34400
5961145 + 1 = 1465000
  • 어떤 차량이 입차된 후에 출차된 내역이 없다면, 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)에 입차되는 경우는 입력으로 주어지지 않습니다.
    • 아래의 예를 포함하여, 잘못된 입력은 주어지지 않습니다.
      • 주차장에 없는 차량이 출차되는 경우
      • 주차장에 이미 있는 차량(차량번호가 같은 차량)이 다시 입차되는 경우

입출력 예

feesrecordsresult
[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

  • 요금표
기본 시간(분)기본 요금(원)단위 시간(분)단위 요금(원)
120060591
  • 입/출차 기록
시각(시:분)차량 번호내역
16:003961입차
16:000202입차
18:003961출차
18:000202출차
23:583961입차
  • 자동차별 주차 요금
차량 번호누적 주차 시간(분)주차 요금(원)
02021200
3961120 + 1 = 1210 +⌈(121 - 120) / 60⌉x 591 = 591
  • 3961번 차량은 2번째 입차된 후에는 출차된 내역이 없으므로, 23:59에 출차되었다고 간주합니다.

입출력 예 #3

  • 요금표
기본 시간(분)기본 요금(원)단위 시간(분)단위 요금(원)
1461110
  • 입/출차 기록
시각(시:분)차량 번호내역
00:001234입차
  • 자동차별 주차 요금
차량 번호누적 주차 시간(분)주차 요금(원)
12341439461 +⌈(1439 - 1) / 1⌉x 10 = 14841
  • 1234번 차량은 출차 내역이 없으므로, 23:59에 출차되었다고 간주합니다.

문제풀이

목표: 주차 요금을 나타내는 정수 배열 fees, 자동차의 입/출차 내역을 나타내는 문자열 배열 records가 매개변수로 주어질 때, 차량 번호가 작은 자동차부터 청구할 주차 요금을 차례대로 정수 배열에 담아서 return

내 풀이

문제를 크게 3단계로 나눠보면 이렇게 나눌 수 있을 것 같다.

  1. records의 시간, 차량번호, 내역을 분리해서 시간은 분 단위로 저장하기.
  2. 내역에 따라서 입/출차 관리
  3. 차량 번호가 작은 자동차부터 주차 요금 계산해서 return

✅ 1번

매개변수로 주어지는 records를 보면 띄어쓰기를 기준으로 나뉘어 있는 것을 알 수 있다. 따라서, for문을 돌며 각각의 기록별로 split()을 하면 시간, 차량번호, 내역을 분리할 수 있을 것이다.

그리고 시간은 시간*60 + 분 으로 다시 저장한다.

def solution(fees, records):
    parking_times = {}  # 입차 시간 기록
    total_times = {}  # 차량별 누적 주차 시간 기록
    
    # 1. records의 시간, 차량번호, 내역을 분리
    for record in records:
        time, car_num, status = record.split()
        
        # 시간은 다시 ':'를 기준으로 분리해서 분 단위로 저장
        h, m = map(int, time.split(':'))
        time = h * 60 + m

✅ 2번

내역이 입차라면 입차 시간 기록 딕셔너리 parking_times 에 ‘차량번호:시간’ 형태로 저장해주고, 출차라면 parking_times에서 입차 시간을 pop 한 다음에 차량별 누적 주차 시간 기록 total_times 에 주차 시간을 계산해서 저장한다.

그리고, records를 다 돌았는데도 출차하지 않는 차량이 있다면 23:59에 출차한 것으로 처리.

# 2. 내역에 따라서 입/출차 관리
        if status == 'IN':  # 들어온 차량이면 딕셔너리에 '차량번호:시간' 형태로 저장
            parking_times[car_num] = time
        else:  
            in_time = parking_times.pop(car_num)  # 출차 시 입차 시각 제거 및 주차 시간 계산
            parked_time = time - in_time
            total_times[car_num] = total_times.get(car_num, 0) + parked_time
    
    # 출차 기록이 없다면 23:59에 출차한 것으로 처리    
    for car_num, in_time in parking_times.items():
        parked_time = (23 * 60 + 59) - in_time
        total_times[car_num] = total_times.get(car_num, 0) + parked_time

✅ 3번

요금 계산을 할 때는 기본 시간을 초과하는 경우와 그렇지 않은 경우로 나눠 계산해야 한다. 이 때 주의할 점은 추가 시간을 단위 시간으로 나눌 때 소수점이 있다면 올림 처리를 해야하는 것.

 # 3. 요금 계산
    total_times = dict(sorted(total_times.items()))  # 차량번호 순으로 정렬
    
    answer = []
    for car_num, time in total_times.items():
        if time <= fees[0]:
            answer.append(fees[1])
        else:
            extra_time = time - fees[0]
            extra_fee = (extra_time + fees[2] - 1) // fees[2] * fees[3]
            answer.append(fees[1] + extra_fee)
    
    return answer

최종 코드

def solution(fees, records):
    parking_times = {}  # 입차 시간 기록
    total_times = {}  # 차량별 누적 주차 시간 기록
    
    # 1. records의 시간, 차량번호, 내역을 분리
    for record in records:
        time, car_num, status = record.split()
        
        # 시간은 다시 ':'를 기준으로 분리해서 분 단위로 저장
        h, m = map(int, time.split(':'))
        time = h * 60 + m
        
        # 2. 내역에 따라서 입/출차 관리
        if status == 'IN':  # 들어온 차량이면 딕셔너리에 '차량번호:시간' 형태로 저장
            parking_times[car_num] = time
        elif status == 'OUT':  
            in_time = parking_times.pop(car_num)  # 출차 시 입차 시각 제거 및 주차 시간 계산
            parked_time = time - in_time
            total_times[car_num] = total_times.get(car_num, 0) + parked_time
    
    # 출차 기록이 없다면 23:59에 출차한 것으로 처리    
    for car_num, in_time in parking_times.items():
        parked_time = (23 * 60 + 59) - in_time
        total_times[car_num] = total_times.get(car_num, 0) + parked_time
    
    # 3. 요금 계산
    total_times = dict(sorted(total_times.items()))  # 차량번호 순으로 정렬
    
    answer = []
    for car_num, time in total_times.items():
        if time <= fees[0]:
            answer.append(fees[1])
        else:
            extra_time = time - fees[0]
            extra_fee = (extra_time + fees[2] - 1) // fees[2] * fees[3]
            answer.append(fees[1] + extra_fee)
    
    return answer
profile
Donghyun
데이터분석 공부 일기~!
이전 포스트

[프로그래머스] k진수에서 소수 개수 구하기 - 파이썬

다음 포스트

[코드카타] 모음 사전 - 파이썬

0개의 댓글