프로그래머스 2022 KAKAO BLIND RECRUITMENT '주차 요금 계산' 파이썬 풀이

프로그래머스

"""
(전략)

주차 요금을 나타내는 정수 배열 fees, 자동차의 입/출차 내역을 나타내는 문자열 배열 records가 
매개변수로 주어집니다. 차량 번호가 작은 자동차부터 청구할 주차 요금을 
차례대로 정수 배열에 담아서 return 하도록 solution 함수를 완성해주세요.
"""

import math
def solution(fees, records):
    sales = {}
    for record in records:
        time, vnum, info = record.split()
        hour, minute = map(int, time.split(':'))
        
        if vnum in sales:
            if info == 'IN':
                sales[vnum][1:] = [hour*60 + minute, 1439]
            else:
                sales[vnum][0] += ((hour*60 + minute) - sales[vnum][1])
                sales[vnum][1:] = [0, 0]
        else:
            sales[vnum] = [0, hour*60 + minute, 1439]
        
        
    result = []
    for vnum, fare in sales.items():
        total = fare[0] + (fare[2] - fare[1])
        extra = math.ceil((total-fees[0])/fees[2])*fees[3]
        fee = fees[1] + (extra if extra > 0 else 0)
        result.append([vnum, fee])
    
    return [v[1] for v in sorted(result, key=lambda x: x[0])]


"""
문제대로 풀어서 카카오에서 원하는 방식인지는 모르겠다.
먼저 records를 보기좋게 바꾸기 위해 for문을 한번 사용했다.

records를 time과 vnum (vehicle number) in/out 여부로 split하였다.
스트링형인 시간을 분으로 전부 맞추기 위해 다시 time을 hour, minute으로 나누었다.

그리고 차량번호(vnum)가 매출장부(sales)에 있을때와 없을때를 체크한다.
if else로 있을때를 먼저 비교하지만, 없을때가 가장 먼저 생기므로
else문부터 설명하면 매출장부에 차량번호를 적는다.
매출장부에는 [몇분동안 주차했는지, 들어온 시간, 나간시간] 으로 적어준다.

처음에는 아직 나가지 않았기 때문에 [0, .., ..] 이고
들어온 시간을 분으로 통일하여 [.., hour*60+minute, ...]으로 적었고,
나간 시간은 문제에서 적혀있지 않다면 default가 23:59이기 때문에
계산하여 [.., .., 1439(23*60+59)]로 적어주었다.

그리고 if문으로 돌아와서 매출장부(sales)에 차량번호가 있다면
들어온건지 나간건지를 확인한다.

들어온거라면 else문에 적은 것과 마찬가지로 적어준다.
이때는 몇분동안 주차했는지 시간이 적혀 있어 [@, hour*60+minute, 1439]로 된다.

나간것이라면 출차시간에서 입차시간을 빼주어 주차한 시간에 넣어주고
[@, 0, 0]으로 세팅해준다.

for문을 다 끝내고 리스트로 만들어주었다.
리스트로 만들때 예외로 입차시간만 있는 경우가 있어
한번더 출차시간에서 입차시간을 빼서 처리를 해주었다.

그리고 fees에서 기본시간과 요금등을 문제에서 나온 식대로 처리하였다.
마지막에 리스트를 차량번호로 소트하여 요금만 리턴시켜주었다.
"""