[프로그래머스] Swift 2022 kakao 주차 요금 계산

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

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

분석 :
string 파싱을 이용한 구현 문제였다. C++에서는 시간을 구하는데 * 60을 하는 등 되게 복잡했는데 swift는 DateFormatter()가 있다 히히 그래서 입차, 출차 시간을 초로 바꾸어 분단위로 누적해주었다.
그리고 out을 할때마다 -1을 해주었는데 -1이 안되었다면 out을 하지 않은 것으로 out하지 않은 차들에 대해서 한번 더 값을 누적시킨뒤 cal() 함수를 통해 주차 요금을 계산하고 차 번호가 작은 순으로 정렬하였다.

Swift 코드 :

//
//  main.swift
//  SOMA👩🏻‍💻
//
//  Created by JoSoJeong on 2022/04/19.
//

import Foundation

func timeToSecond(_ s: String) -> Double {
    let dateFormatter = DateFormatter()
    dateFormatter.dateFormat = "HH:mm"
    return Double(dateFormatter.date(from: s)!.timeIntervalSince1970)
}

func cal(_ fees: [Int], _ now: Double) -> Int {
    let defaultCost: Int = fees[1]
    let defaultTime: Double = Double(fees[0])
    if now < defaultTime { return defaultCost }
    let div: Double = Double(fees[2])
    let mul: Double = Double(fees[3])
    return defaultCost + Int(ceil((now - defaultTime) / div) * mul)
}
               
func solution(_ fees:[Int], _ records:[String]) -> [Int] {
    var answer = [(String, Int)]() // number : cost
    var stack: [String : Double] = [:] // 입차 출차 시간 계산
    var dic: [String: Double] = [:] // car number - 누적 시간

    for i in records {
        let line = i.split(separator: " ").map { String($0) }
        let secondT: Double = timeToSecond(line[0])
        let number = line[1]

        if (stack[number] != 0) && (line[2] == "IN") {
            // 처음 입차
            stack[number] = secondT
        }else if line[2] == "OUT" {
            // 시간 계산
            // stack pop
            let last = stack[number]!
            stack[number] = -1
            // dic에 누적
            dic[number, default: 0] += (secondT - last) / 60  //출차시간 - 입차시간
        }
    }
    
    for i in stack { //stack이 비지 않았다면 입차는 했으나 출차하지 않은 것
        if i.value != -1 { // 아직 빠져나가지 x
            let lastT = timeToSecond("23:59")
            dic[i.key, default: 0] += (lastT - i.value) / 60
        }
    }
    
    for i in dic {
        let cost = cal(fees, i.value)
        answer.append((i.key, cost))
    }
    
    answer.sort { $0.0 < $1.0 }
    return answer.map { $0.1 }
    
}

결과 :