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

문제 링크: https://school.programmers.co.kr/learn/courses/30/lessons/92341

프로그래머스

코드

import math
def solution(fees, records):
    dict = {}
    maxt = (23*60)+59
    for i in records:
        tmp = i.split()
        min,sec = tmp[0].split(":")
        if tmp[2] == 'IN':
            if tmp[1] in dict:
                dict[tmp[1]] += maxt - (int(min)*60+int(sec))
            else:
                dict[tmp[1]] = maxt- (int(min)*60+int(sec))
        elif tmp[2] == 'OUT':
            buf = dict[tmp[1]]
            a = maxt - (int(min)*60+int(sec))
            dict[tmp[1]] = buf - a    
    arr = list(dict.items())
    answer = []
    for i in range(len(arr)):
        if arr[i][1] <= fees[0]: 
            answer.append([int(arr[i][0]),fees[1]])
        else:
            answer.append([int(arr[i][0]),fees[1] + math.ceil((arr[i][1]-fees[0])/fees[2])*fees[3]])
    answer.sort(key=lambda x: x[0])
    result = []
    for x in answer:
        result.append(x[1])
    return result

풀이

• 더 간결하고 좋은 코드들이 있을꺼라 생각하지만 나의 코드를 해석해보면,

import math
def solution(fees, records):
    dict = {}
    maxt = (23*60)+59
    for i in records:
        tmp = i.split()
        min,sec = tmp[0].split(":")
        if tmp[2] == 'IN':
            if tmp[1] in dict:
                dict[tmp[1]] += maxt - (int(min)*60+int(sec))
            else:
                dict[tmp[1]] = maxt- (int(min)*60+int(sec))
        elif tmp[2] == 'OUT':
            buf = dict[tmp[1]]
            a = maxt - (int(min)*60+int(sec))
            dict[tmp[1]] = buf - a    
    arr = list(dict.items())

• 출차 최대 시간을 maxt로 하고, 딕셔너리에 넣을때 입차시간 기준 최대 이용 시간(maxt-입차시간)을 넣어주었다.
• 출차할 경우 딕셔너리 값을 갱신할 때, 이용 시간을 넣기 위해서 입차할때 넣었던 최대 이용시간 - (maxt-출차시간)으로 갱신해준다.
• 만약 두번째 입차인경우 기존 이용 시간에, 입차시간 기준 최대 이용시간(maxt-입차시간)을 갱신하지 않고 더해주어야 한다.
• 마지막엔 딕셔너리의 키,밸류를 튜플로 한 리스트를 만들어준다.(번호판이 낮은 수부터 정렬하기 위함)

• 이용 요금을 계산하는 코드에는 조건문을 이용해서 기본 시간 이하이면 기본요금, 이상이면 값을 계산했는데, 이 때 올림을 위해 math.ceil()을 사용했다.