[프로그래머스] / [Level 2] / [Python] 주차 요금 계산

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

프로그래머스

---

오랜만에 다시 풀었는데 예전 코드랑 거의 비슷해서 신기했다.

from collections import defaultdict
from collections import deque
import math
def solution(fees, records):
    answer = []
    
    rd = defaultdict(list) # records dict
    for record in records:
        t, n, s = record.split(' ')        
        rd[n].append([(int(t[:2])*60 + int(t[3:])), s])
    
    rd = dict(sorted(rd.items()))

    for k, v in rd.items():
        inout = deque()
        value = 0
        for v_ in v:
            if v_[1] == 'IN':
                inout.append(v_[0])
            else:
                value += v_[0] - inout.popleft()        
        else:
            if inout:
                value += 1439 - inout.popleft()
        
        if value <= fees[0]:
            answer.append(fees[1])
        else:
            answer.append(fees[1] + math.ceil((value-fees[0])/fees[2])*fees[3])
            
    return answer