[프로그래머스] lv.2 주차 요금 계산 python

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

프로그래머스

문제 설명

주차장의 요금표와 차량이 들어오고(입차) 나간(출차) 기록이 주어졌을 때, 차량별로 주차 요금을 계산하려고 합니다. 아래는 하나의 예시를 나타냅니다.

어떤 차량이 입차된 후에 출차된 내역이 없다면, 23:59에 출차된 것으로 간주합니다.
0000번 차량은 18:59에 입차된 이후, 출차된 내역이 없습니다. 따라서, 23:59에 출차된 것으로 간주합니다.
00:00부터 23:59까지의 입/출차 내역을 바탕으로 차량별 누적 주차 시간을 계산하여 요금을 일괄로 정산합니다.
누적 주차 시간이 기본 시간이하라면, 기본 요금을 청구합니다.
누적 주차 시간이 기본 시간을 초과하면, 기본 요금에 더해서, 초과한 시간에 대해서 단위 시간 마다 단위 요금을 청구합니다.
초과한 시간이 단위 시간으로 나누어 떨어지지 않으면, 올림합니다.
⌈a⌉ : a보다 작지 않은 최소의 정수를 의미합니다. 즉, 올림을 의미합니다.
주차 요금을 나타내는 정수 배열 fees, 자동차의 입/출차 내역을 나타내는 문자열 배열 records가 매개변수로 주어집니다. 차량 번호가 작은 자동차부터 청구할 주차 요금을 차례대로 정수 배열에 담아서 return 하도록 solution 함수를 완성해주세요.

 

제한 사항

fees의 길이 = 4
fees[0] = 기본 시간(분)
1 ≤ fees[0] ≤ 1,439
fees[1] = 기본 요금(원)
0 ≤ fees[1] ≤ 100,000
fees[2] = 단위 시간(분)
1 ≤ fees[2] ≤ 1,439
fees[3] = 단위 요금(원)
1 ≤ fees[3] ≤ 10,000
1 ≤ records의 길이 ≤ 1,000

records의 각 원소는 "시각 차량번호 내역" 형식의 문자열입니다.
시각, 차량번호, 내역은 하나의 공백으로 구분되어 있습니다.
시각은 차량이 입차되거나 출차된 시각을 나타내며, HH:MM 형식의 길이 5인 문자열입니다.
HH:MM은 00:00부터 23:59까지 주어집니다.
잘못된 시각("25:22", "09:65" 등)은 입력으로 주어지지 않습니다.
차량번호는 자동차를 구분하기 위한, `0'~'9'로 구성된 길이 4인 문자열입니다.
내역은 길이 2 또는 3인 문자열로, IN 또는 OUT입니다. IN은 입차를, OUT은 출차를 의미합니다.
records의 원소들은 시각을 기준으로 오름차순으로 정렬되어 주어집니다.
records는 하루 동안의 입/출차된 기록만 담고 있으며, 입차된 차량이 다음날 출차되는 경우는 입력으로 주어지지 않습니다.
같은 시각에, 같은 차량번호의 내역이 2번 이상 나타내지 않습니다.
마지막 시각(23:59)에 입차되는 경우는 입력으로 주어지지 않습니다.
아래의 예를 포함하여, 잘못된 입력은 주어지지 않습니다.
주차장에 없는 차량이 출차되는 경우
주차장에 이미 있는 차량(차량번호가 같은 차량)이 다시 입차되는 경우

 

문제 풀이

먼저 fees로 들어오는 기본 시간, 기본 요금, 단위 시간, 단위 요금을 각각 t,f,dt,df로 받아줍니다.

그리고 dictionary를 이용해서 records로 들어오는 자동차의 시간(time), 차번호(number), 내역(IO)을 저장했습니다.

여기서 시간을 저장할 때 60분법으로 바꿔서 저장을 해주었는데 그 이유는 시간 계산할 때 복잡함을 줄이려고 했습니다.

def solution(fees, records):
    t,f,dt,df=fees
    
    answer = []
    dic=dict()
    for i in range(len(records)):
        time,number,IO=records[i].split()
        h,m=time.split(":")
        time=int(h)*60+int(m)
        #print(time)

만약 dictionary에 차번호가 있다면 시간과 내역을 넣고,

없다면 새로 차번호를 넣고 그 안에 시간과 내역을 넣도록 했습니다.

if number in dic:
    dic[number].append([time,IO])
else:
    dic[number]=[[time,IO]]

이렇게 정리된 값들을 list로 바꿔서 result란 변수에 담아주고 정렬한다.

리스트로 정리되었으므로 편하게 반복문을 돌릴 수 있다.

 

result 하나하나를 item이란 이름으로 가져오고 그 안에 담긴 내역이 IN이 라면 imsi에 시간(con[0])을 빼주고 OUT이라면 시간을 더해준다.

만약 마지막 내역이 IN이 라면 imsi에 23시 59분을 60분법으로 바꾼 값을 더해준다.

 

이렇게 계산된 imsi가 기본 시간(t)보다 작다면 answer에 그냥 append해주고

만약 더 크다면 단위시간(dt)과 단위요금(df)도 계산한 후 answer에 append해주면 된다 !

result=list(dic.items())
result.sort(key=lambda x:x[0])
print(result)
for item in result:
    imsi=0
    for con in item[1]:
        if con[1]=='IN':
            imsi -= con[0]
        else:
            imsi += con[0]
    if item[1][-1][1] =='IN':
        imsi += 23*60+59


    if imsi <= t:
        answer.append(f)
    else:
        answer.append(f+math.ceil((imsi-t)/dt)*df)

그래서 모든 코드를 더한 결과는 다음과 같다.

import math
def solution(fees, records):
    t,f,dt,df=fees
    
    answer = []
    dic=dict()
    for i in range(len(records)):
        time,number,IO=records[i].split()
        h,m=time.split(":")
        time=int(h)*60+int(m)
        #print(time)
        
        if number in dic:
            dic[number].append([time,IO])
        else:
            dic[number]=[[time,IO]]
    
    #print(dic)
    result=list(dic.items())
    result.sort(key=lambda x:x[0])
    print(result)
    for item in result:
        imsi=0
        for con in item[1]:
            if con[1]=='IN':
                imsi -= con[0]
            else:
                imsi += con[0]
        if item[1][-1][1] =='IN':
            imsi += 23*60+59
    
    
        if imsi <= t:
            answer.append(f)
        else:
            answer.append(f+math.ceil((imsi-t)/dt)*df)
        
    
    return answer

 

카카오는 문제가 너무 길어서 힘들어요 ㅠㅠ