[Python] 프로그래머스 코딩테스트 연습 level 2 (주차 요금 계산)

1. 문제 설명

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

 

 

 

 

2. 제한 조건

• 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)에 입차되는 경우는 입력으로 주어지지 않습니다.
  • 아래의 예를 포함하여, 잘못된 입력은 주어지지 않습니다.
    • 주차장에 없는 차량이 출차되는 경우
    • 주차장에 이미 있는 차량(차량번호가 같은 차량)이 다시 입차되는 경우

 

 

 

 

3. 내 풀이

from datetime import datetime
import time
from collections import defaultdict
import math
def solution(fees, records):
    answer = []
    cars = defaultdict(list)
    for i in records:
        car_info = i.split(' ')
        cars[car_info[1]].append(car_info[0]+' '+car_info[2])
    cars = dict(sorted(cars.items()))
    
    for i in cars:
        i_fee = []
        for j in range(len(cars[i])):
            if 'IN' in cars[i][j]:
                intime = datetime.strptime(cars[i][j].split(' ')[0],'%H:%M')
                try:
                    outtime=datetime.strptime(cars[i][j+1].split(' ')[0],'%H:%M')
                except:
                    outtime = datetime.strptime('23:59','%H:%M')
                i_fee.append((outtime-intime).seconds//60)
        
        if sum(i_fee)<fees[0]:
            answer.append(fees[1])
        else:
            answer.append(int(fees[1]+math.ceil((sum(i_fee)-fees[0])/fees[2])*fees[3]))
    
    return answer

 

1. 차량 번호 별 입차 시간과 출차 시간을 받기 위해 defaultdict(list)로 cars 딕셔너리를 생성

 -> 차량 번호 별 [입차, 출차] 딕셔너리 선언

2. cars 딕셔너리를 key값을 기준으로 정렬

3. for 문을 사용하여 차량 별 주차 시간의 합계산

-> IN 글자가 문자열 안에 존재하면 strptime()함수를 사용하여 시간 데이터로 변환

-> try~except 문을 사용하여 입차 다음 출차 기록이 존재하면 출차시간으로 outtime 선언, 출차 기록이 존재하지 않는다면 23:59로 선언

4. 입차시간과 출차시간의 차를 구한 다음 분으로 변환함

5. 차량 번호별 총 주차시간의 합을 구하여 기본 요금 시간보다 적으면 기본요금으로 answer list에 추가

기본 요금 시간보다 많으면 단위시간으로 나눈 뒤, 단위시간별 요금 곱함 + 기본요금 계산하여 answer list에 추가하여  answer 반환

 

 

 

 

4. 다른 사람의 풀이

from collections import defaultdict
from math import ceil

class Parking:
    def __init__(self, fees):
        self.fees = fees
        self.in_flag = False
        self.in_time = 0
        self.total = 0

    def update(self, t, inout):
        self.in_flag = True if inout=='IN' else False
        if self.in_flag:  self.in_time = str2int(t)
        else:             self.total  += (str2int(t)-self.in_time)

    def calc_fee(self):
        if self.in_flag: self.update('23:59', 'out')
        add_t = self.total - self.fees[0]
        return self.fees[1] + ceil(add_t/self.fees[2]) * self.fees[3] if add_t >= 0 else self.fees[1]

def str2int(string):
    return int(string[:2])*60 + int(string[3:])

def solution(fees, records):
    recordsDict = defaultdict(lambda:Parking(fees))
    for rcd in records:
        t, car, inout = rcd.split()
        recordsDict[car].update(t, inout)
    return [v.calc_fee() for k, v in sorted(recordsDict.items())]

 

 

 

import math

def solution(fees, records):
    check = {}

    for record in records:
        time, number, status = record.split()
        time = time.split(':')
        time = int(time[0])*60 + int(time[1])
        if number not in check:
            check[number] = (0, time, status)
        if status == 'IN':
            check[number] = (check[number][0], time, status)
        elif status == 'OUT':
            total_time, in_time, _ = check[number]
            total_time += time - in_time
            check[number] = (total_time, time, status)

    result = {}

    for number in check.keys():
        total_time, time, status = check[number]
        if status == 'IN':
            total_time += 1439 - time
        fee = fees[1]
        if total_time <= fees[0]:
            result[number] = fee
        else:
            fee = fee + math.ceil((total_time - fees[0]) / fees[2]) * fees[-1]
            result[number] = fee

    return list(map(lambda x : x[1], sorted(result.items())))