[Python] 프로그래머스 Lv2 - 주차 요금 계산 (2022 KAKAO BLIND RECRUITMENT)

🔗 문제 링크

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

📂 분류

구현

💡 풀이

딕셔너리 사용법이 익숙하지 않아 고생했지만 단순 구현 문제로 쉽게 해결할 수 있는 문제이다.

 

접근 방식

• 딕셔너리를 사용해서 차번호 별로 입/출차 기록을 저장한다.
• 입/출차 시간을 계산하기 위해 분 단위로 환산하여 딕셔너리에 저장한다.
• stack을 이용해서 총 주차 시간을 계산한다.
  • 마지막 출차 기록이 없다면 23:59에 출차된 것으로 간주한다고 나와있다.
  • 따라서 stack이 비어있지 않을 경우, LAST_EXIT_TIME에서 빼주어 총 주차 시간에 더해준다.

💻 코드

from math import ceil

LAST_EXIT_TIME = 1439


def hourToMinutes(time):
    h, m = map(int, time.split(':'))
    return h * 60 + m


def solution(fees, records):
    answer = []
    car = {}
    dt, df, ut, uf = fees

    for r in records:
        time, number, op = r.split()

        if number in car:
            car[number].append((hourToMinutes(time), op))
        else:
            car[number] = [(hourToMinutes(time), op)]

    car_list = list(car.items())
    car_list.sort(key=lambda x: x[0])

    for cars in car_list:
        time = 0
        stack = []

        for car in cars[1]:
            if car[1] == 'IN':
                stack.append(car[0])
            else:
                time += car[0] - stack.pop()

        if len(stack) != 0:
            time += LAST_EXIT_TIME - stack.pop()

        if time <= dt:
            answer.append(df)
        else:
            answer.append(df + ceil((time - dt) / ut) * uf)

    return answer