[프로그래머스] 2022 KAKAO BLIND RECRUITMENT - 주차 요금 계산

문제

문제 설명

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

요금표

입/출차 기록

자동차별 주차 요금

• 어떤 차량이 입차된 후에 출차된 내역이 없다면, 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 : [180, 5000, 10, 600]
records : ["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"]
result : [14600, 34400, 5000]
-----------------------------
fees : [120, 0, 60, 591]
records : ["16:00 3961 IN","16:00 0202 IN","18:00 3961 OUT","18:00 0202 OUT","23:58 3961 IN"]	[0, 591]
[1, 461, 1, 10]	["00:00 1234 IN"]
result : [14841]
'''

records에서 각각 문자열 요소를 시간 / 차 번호 / 입,출차여부 로 나눠주기로 했다.

그리고 차 번호를 key 값으로 가지고 시간,상태를 value로 가지는 dictionary 를 사용하자.

해당 딕셔너리를 temp_dict라고 하고, split을 이용해서 공백을 기준으로 자른다.

temp_dict = {} # 입, 출차 정보 담는 딕셔너리
    for record in records:
        time,number,status = record.split()

그리고 시간을 계산해야 하니까 00시를 기준으로 분단위로 나타내주기로 했다. 그리고 출 - 입 하면 되겠지?

time_dict = {} # 해당 차의 주차시간을 담는 딕셔너리

hour,minute = time.split(':')
minutes = int(hour)*60 + int(minute)

여기서 또 문제가 생겼다. status가 'IN' 일 때와 'OUT' 일 때를 나눠서 저장하고 시간을 계산하려고 했는데,

첫째로 OUT 이 없는 경우를 정해줘야 한다. 23시59분으로...

둘째 IN 이 여러번인 경우도 있다.

그리고 둘 다 섞인 경우도.... 그래서 생각한 것은

1. 기존 딕셔너리에 모든 걸 저장하지 않고, '차번호' : '분단위 시각' 의 형태로 시간만 저장한다.

2. 시간을 담는 딕셔너리를 새로 만들어서, OUT 일 때는 출차시각-입차시각 의 결과를 시간딕셔너리에 저장

3. OUT 까지 완료되면 기존 딕셔너리에서 삭제

그렇게 하면 OUT 된 시간은 차곡차곡 시간 딕셔너리에 담기고, OUT되지 않은 시간은 그대로 기존 딕셔너리에 남게 된다.

if status == 'IN': # 입차일때
    temp_dict[number] = minutes # '차번호' : '분단위 시각'
elif status == 'OUT': # 출차일때
    time = minutes - temp_dict[number] # 출차시각-입차시각을 시간 딕셔너리에 저장
    if number in time_dict:
        time_dict[number] += time # 기존 키값이 있다면 새 요소로 저장
    else:
        time_dict[number] = time # 기존에 없으면 초기화하면서 저장
    del(temp_dict[number]) # OUT 까지 완료되면 기존 딕셔너리에서 삭제

이제 기존 딕셔너리엔 출차가 찍히지 않은 차의 정보만 남았으니까, 

이 값들을 23시59분에 출차했다고 계산해주면 된다.

for number, val in temp_dict.items(): # OUT되지 않은 차의 정보만 남은 딕셔너리
    time = 1439 -val # 23시59분 출차로 계산한다.
    if number in time_dict:
        time_dict[number] += time # 기존 키값이 있다면 새 요소로 저장
    else:
        time_dict[number] = time # 기존에 없으면 초기화하면서 저장

시간 계산이 끝났으니 이걸 가지고 주차비만 계산해주면 된다.

결국 주차비는 식 하나로 표현이 되는데,

기본요금 + ((누적주차시간-기본시간)/단위시간)올림 * 단위요금

이므로 누적 주차 시간이 기본 시간을 초과하는지 여부만 확인해주면 된다.

# 주차비 계산        
    for number, time in time_dict.items(): 
        over_time = max(0, time-fees[0]) # 기본요금 초과시간
        total_fees = fees[1] + math.ceil(over_time/fees[2]) * fees[3] # 주차비 계산
        time_dict[number] = total_fees # 기존 시간 딕셔너리 사용해서 시간을 주차비로 업데이트
    time_dict = sorted(time_dict.items())

 

전체 코드

import math

'''
주차 요금 계산
주차 요금을 나타내는 정수 배열 fees, 
자동차의 입/출차 내역을 나타내는 문자열 배열 records
차량 번호가 작은 자동차부터 청구할 주차 요금을 차례대로 정수 배열에 담아서 return
'''

# 1. 분류 - 완료
# 2. 주차시간 계산
# 3. 23:59 출차 적용
# 4. 요금 계산
# 5. 정렬

# 기본요금 + ((누적주차시간-기본시간)/단위시간)올림 * 단위요금
# 시간으로 나타내야 함,, 00시를 기준으로??? 분단위로? ex 05:34 5 * 60 + 34
# 출차시간 - 입차시간

def solution(fees, records):
    answer = []
    temp_dict = {} # 입, 출차 정보 담는 딕셔너리
    time_dict = {} # 해당 차의 주차시간을 담는 딕셔너리
    for record in records:
        time,number,status = record.split() # 시간 / 차 번호 / 입,출차여부
        hour,minute = time.split(':')
        minutes = int(hour)*60 + int(minute) # 00시 기준으로 분단위 시각

        if status == 'IN': # 입차일때
            temp_dict[number] = minutes # '차번호' : '분단위 시각'
        elif status == 'OUT': # 출차일때
            time = minutes - temp_dict[number] # 출차시각-입차시각을 시간 딕셔너리에 저장
            if number in time_dict:
                time_dict[number] += time # 기존 키값이 있다면 새 요소로 저장
            else:
                time_dict[number] = time # 기존에 없으면 초기화하면서 저장
            del(temp_dict[number]) # OUT 까지 완료되면 기존 딕셔너리에서 삭제
    # 출차 없을 때            
    for number, val in temp_dict.items(): # OUT되지 않은 차의 정보만 남은 딕셔너리
        time = 1439 -val # 23시59분 출차로 계산한다.
        if number in time_dict:
            time_dict[number] += time # 기존 키값이 있다면 새 요소로 저장
        else:
            time_dict[number] = time # 기존에 없으면 초기화하면서 저장
    # 주차비 계산        
    for number, time in time_dict.items(): 
        over_time = max(0, time-fees[0]) # 기본요금 초과시간
        total_fees = fees[1] + math.ceil(over_time/fees[2]) * fees[3] # 주차비 계산
        time_dict[number] = total_fees # 기존 시간 딕셔너리 사용해서 시간을 주차비로 업데이트
    time_dict = sorted(time_dict.items())

    for number, fee in time_dict:
        answer.append(fee) 
        
    return answer   
       
       
# 입출력 예
'''
fees : [180, 5000, 10, 600]
records : ["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"]
result : [14600, 34400, 5000]
-----------------------------
fees : [120, 0, 60, 591]
records : ["16:00 3961 IN","16:00 0202 IN","18:00 3961 OUT","18:00 0202 OUT","23:58 3961 IN"]	[0, 591]
[1, 461, 1, 10]	["00:00 1234 IN"]
result : [14841]
'''

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"]))