프로그래머스 주차 요금 계산 Kotlin (문자열, 시간)

문제 출처 : https://programmers.co.kr/learn/courses/30/lessons/92341

코딩테스트 연습 - 주차 요금 계산

문제 설명

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

제한사항

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

풀이

이런 시간 문제는 항상 시간 단위를 통일하고 시작하자.

이 문제에선 최소 시간 단위가 분이므로 분 단위로 통일하고, 다음 날로 넘어가진 않고 00:00 ~ 23:59의 시간이 주어지므로 그렇게 어렵지 않다. 잘못된 입력도 주어지지 않으니 얼마나 친절한가.

 

음.. 단순히 문제에서 요구하는 것들을 구현하면 되는 문제라 딱히 설명할 건 없고,

핵심은 해시와 시간 단위 통일이다.

대략적인 흐름은, 입차, 출차를 모두 map으로 입력받아서 입차인 경우 삽입, 출차인 경우 삭제(이때 금액 계산하지 않고 주차 시간만 누적한다.) 최종적으로 남아있는 차들에 대해서 주차 시간 더해주기. 이후 차들의 주차 시간들로 요금을 계산하고 정렬하여 출력하면 끝.

코드

import java.util.*

class Solution {
    var pTime = 0
    var pCost = 0
    var aTime = 0
    var aCost = 0

    val inMap = mutableMapOf<String, Int>()
    val resultTime = TreeMap<String, Int>()

    fun calCost(t: Int): Int{
        var cost = pCost
        var time = t
        time -= pTime
        //기본 시간 이상이라현 더 청구
        if(time>0){
            var overTime = if(time%aTime==0) time/aTime else time/aTime+1
            cost += overTime*aCost
        }
        return cost
    }

    fun solution(fees: IntArray, records: Array<String>): IntArray {

        //preset
        pTime = fees[0]
        pCost = fees[1]
        aTime = fees[2]
        aCost = fees[3]


        for (record in records) {
            val (tt, car, action) = record.split(' ')
            var t = 0
            tt.split(':').map { it.toInt() }.apply {
                t += this[0]*60 + this[1]
            }

            if(action=="IN"){
                inMap[car]=t
            }
            else{
                resultTime[car]= resultTime.getOrDefault(car,0)+t-inMap[car]!!
                inMap.remove(car)
            }

        }
        inMap.forEach { car, t ->
            resultTime[car]= resultTime.getOrDefault(car,0)+(23*60 + 59)-t
        }
        var answer = IntArray(resultTime.size)
        var idx=0

        resultTime.forEach { s, i ->
            answer[idx++] = calCost(i)
        }

        return answer
    }
}