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

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

프로그래머스

---

python

import math
from collections import deque

def calc_time(car, carnum, out_h, out_m):
    in_h, in_m = int(car[carnum][0]), int(car[carnum][1])
    if in_m > out_m:
        out_h -= 1
        out_m += 60
    time = (out_m - in_m) + 60*(out_h - in_h)
    return time

def solution(fees, records):
    answer = dict()
    car = dict()
    q = deque()
    for info in records:
        h, m, carnum, state = int(info[:2]), int(info[3:5]), info[6:10], info[11:]
        if state == "IN":
            car[carnum] = (h,m)
            q.append(carnum)
        else:
            if carnum not in answer:
                answer[carnum] = calc_time(car, carnum, h, m)
            else:
                answer[carnum] += calc_time(car, carnum, h, m)
            q.remove(carnum)
            
    while q:
        carnum = q.popleft()
        if carnum not in answer:
            answer[carnum] = calc_time(car, carnum, 23, 59)
        else:
            answer[carnum] += calc_time(car, carnum, 23, 59)
            
    for i in answer:
        if answer[i] <= fees[0] :
            answer[i] = fees[1]
        else:
            answer[i] = fees[1] + math.ceil((answer[i]-fees[0])/fees[2]) * fees[3]
    answer = sorted(answer.items())
    return [i[1] for i in answer]

푸는데 한시간에서 한시간 반정도 걸렸다..

엄청 까다롭지도 않았던 것 같은데 시간은 드럽게 오래 걸린 문제 ㅠㅠ

구현문제 많이 풀어보다보면 좀 늘겠지?

 

제출하고 나서 다른 분들 풀이를 많이 읽어봤는데, 읽어보면서 가장 크게 느낀 점은

1. h, m, carnum, state 인자를 받을 때 굳이 리스트 슬라이싱 할 필요 없이 split() 함수를 사용하면 됐다

2. 출차시간 - 입차시간 계산할 때 시간, 분 따로따로 복잡하게 계산할 필요가 없었다

---

아주 미세하게 쪼금 더 보기 좋게 수정해 보았다.

import math
from collections import deque

def calc_time(car, carnum, out_h, out_m):
    in_h, in_m = int(car[carnum][0]), int(car[carnum][1])
    return (out_h*60 + out_m) - (in_h*60 + in_m)

def solution(fees, records):
    answer = dict()
    car = dict()
    q = deque()
    for info in records:
        h, m, carnum, state = info.replace(":"," ").split()
        h, m, carnum = map(int,[h, m, carnum] )
        if state == "IN":
            car[carnum] = (h,m)
            q.append(carnum)
        else:
            if carnum not in answer:
                answer[carnum] = calc_time(car, carnum, h, m)
            else:
                answer[carnum] += calc_time(car, carnum, h, m)
            q.remove(carnum)
            
    while q:
        carnum = q.popleft()
        if carnum not in answer:
            answer[carnum] = calc_time(car, carnum, 23, 59)
        else:
            answer[carnum] += calc_time(car, carnum, 23, 59)
            
    for i in answer:
        if answer[i] <= fees[0] :
            answer[i] = fees[1]
        else:
            answer[i] = fees[1] + math.ceil((answer[i]-fees[0])/fees[2]) * fees[3]
    answer = sorted(answer.items())
    return [i[1] for i in answer]