상상쓰

[프로그래머스] 주차 요금 계산 본문

Coding Test

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

상상쓰 2022. 2. 12. 00:41

https://programmers.co.kr/learn/courses/30/lessons/92341

 

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

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

programmers.co.kr

 

오랜만에 풀어서 그런지 문법들이 기억이 잘 안 난다.

dictionary 를 이용하여 입차 시간과 출차 될 때 주차시간을 기록한다. 

 

import math

def solution(fees, records):
    answer = []
    dic = {}
    last_time = 23 * 60 + 59
    
    for i in records:
        time, car, history = i.split(' ')
        h, m = time.split(':')
        time = int(h) * 60 + int(m)
        
        if history == 'IN':
            if car not in dic:
                dic[car] = [time, 0]
            else:
                dic[car] = [time, dic[car][1]]
        else:
            dic[car] = [-1, dic[car][1] + time - dic[car][0]]
            
    cars = sorted(dic.keys())
    
    for i in cars:
        mm = dic[i][1]
        
        if dic[i][0] >= 0:
            mm = mm + last_time - dic[i][0]
        
        answer.append(fees[1] + math.ceil(max(0, mm - fees[0]) / fees[2]) * fees[3])

    return answer

print(solution([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]
Comments