[Programmers] Lv2. 주차 요금 계산(kotlin)

문제

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

 

 

 

예시

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

 

 

 

풀이

• 두 개의 map을 통해 하나는 입/출차 기록과 하나는 해당 차량의 전체 시간을 기록하시면 됩니다.

import kotlin.math.*

class Solution {
    fun solution(fees: IntArray, records: Array<String>): IntArray {
        val score = mutableMapOf<String, Int>()
        var map = mutableMapOf<String, Int>()
        
        records.forEach{ record ->
            val rl = record.split(" ")
            val time = rl.get(0).split(":").let{ it.first().toInt()* 60 + it.last().toInt()}
            val num = rl.get(1)
            
            if(map.contains(num)){
                score.put(num, (score.get(num)?: 0) + time - map.get(num)!!)
                map.remove(num)
            }else{
                map.put(num, time)
            }
        }
        
        for(elem in map){
            score.put(elem.key, (score.get(elem.key)?: 0) + 60*23 + 59 - elem.value)
        }
        
        return score.toSortedMap().map{
            if(it.value <=fees.get(0).toInt()){
                fees.get(1).toInt()
            }else{
                fees.get(1).toInt() + ceil((it.value - fees.get(0).toInt())/fees.get(2).toFloat()).toInt()*fees.get(3).toInt()
            }
        }.toIntArray()
    }
}

 

 

 

참고

• 함수형 프로그래밍 언어인 kotlin의 특성에 맞게 함수를 많이 익히자 😽
• mutableMapOf는 Java의 LinkedHashMap을 생성합니다. LinkedHashMap은 HashMap과 key와 value를 가지고 있는 형태는 동일하나 데이터의 순서를 보장받기 위해서 별도의 Double Linked List를 관리하고 있습니다.
• https://developer.android.com/reference/java/util/LinkedHashMap

LinkedHashMap  |  Android Developers