[프로그래머스 92341] 주차 요금 계산 swift

프로그래머스

문제 설명

 

 

문제 풀이

충실히 구현하면 되는 문제다.

 

유의해야 할 점은 

 

1. 마지막에 나가지 않은 차는 23:59분에 출차한 것으로 간주한다.

2. 초과시간 계산할 때 올림한다. ex) 40분 초과했는데 30분마다 1000원 추가면 2000원 추가

 

딕셔너리를 사용했다.

 

[Int: (isOut: Bool, time: Int)]

[차번호: (나갔는지, 사용시간)]

 

마지막에 나갔는지 체크하기 위해 딕셔너리의 value에 isOut을 추가한 형태다.

 

records를 다 체크하고 만약에 아직 안나간 차가 있다면(isOut이 false이면) 

23:59분에 나갔다고 간주하고 시간을 계산한다.

 

그 다음엔 차번호를 기준으로 오름차순 정렬해주고 요금 계산해주면 된다.

 

 

시간은 분으로 계산하는 메소드를 사용했다.

ex) 01:10이면 70 리턴해줌

 

 

코드

import Foundation

func solution(_ fees:[Int], _ records:[String]) -> [Int] {
    
    // 시간을 분으로 바꿔주는 메소드 ex) 01:10 -> 70
    func timeToMinute(_ time: String) -> Int {
        let split = time.split(separator: ":").map {Int(String($0))!}
        return 60*split[0] + split[1]
    }
    
    // [차번호: (나갔는지, 사용시간)]
    var dict: [Int: (isOut: Bool, time: Int)] = [:]
    
    for record in records {
        let split = record.split(separator: " ").map {String($0)}
        let time = split[0]
        let carNumber = Int(split[1])!
        
        // 차가 들어오면
        if split[2] == "IN" {
            if dict[carNumber] == nil {
                dict[carNumber] = (false, timeToMinute(time))
            }
            else {
                let usageTime = dict[carNumber]!.time + timeToMinute(time)
                dict[carNumber] = (false, usageTime)
            }
        }
        
        // 차가 나가면 isOut true로 변경
        else if split[2] == "OUT" {
            let usageTime =  dict[carNumber]!.time - timeToMinute(time)
            dict[carNumber] = (true, usageTime)
        }
    }
    
    // 아직 안나간 차가 있다면 23:59분에 나갔다고 가정
    for (key, value) in dict {
        if !dict[key]!.isOut {
            let usageTime = dict[key]!.time - timeToMinute("23:59")
            dict[key] = (false, usageTime)
        }
    }
    
    // 차 번호 기준으로 오름차순 정렬
    let dictSorted = dict.sorted { $0.key < $1.key }
    var answer: [Int] = []
    
    for (key, value) in dictSorted {
        let time = value.time*(-1)
        
        // 기본 시간을 넘기지 않았으면 기본 요금만
        if time <= fees[0] {
            answer.append(fees[1])
        }
        
        // 기본 시간을 넘겼다면
        else {
            let c = ceil((Float(time)-Float(fees[0])) / Float(fees[2]))
            let money = fees[1] + Int(c)*fees[3]
            answer.append(money)
        }
    }
    return answer
}

 

 

후기

풀이시간: 1시간

 

그냥 구현하면 되는 문제인데 오래 걸렸다..

 

나누고 올림은 ceil, 반올림 round, 내림은 floor 외우자