[프로그래머스] (Swift) Kakao 주차 요금 계산

 

문제링크

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

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

 

문제 해석 및 풀이

• fees 에서 첫번째는 기본 주차 시간, 두번째는 기본 주차 요금, 세번째는 주차요금을 계산하는 단위 (분), 네번째는 단위당 얼마?를 의미한다.
• records에서 "__" 이 한 따옴표를 " " 공백으로 나누어 번호판과 입출차시간을 구별해주는 것이 필요하다.
  • 생각해보면 IN 과 OUT 은 그닥 중요하지 않다. 어차피 records는 시간순서대로 정렬되어 주어지기 때문에, IN다음은 무조건 OUT일 것이다. 이 개념대로 생각해보면, records에서 번호판 0000의 입출차시간 기록 갯수가 홀수개이면, OUT 기록이 없는 것이라고 판단할 수 있다. 그럼 갯수가 홀수인 것에만 append("23:59")를 해주면 될 것이다!
• 주의해야할 조건 중 하나가, 번호판 오름차순 순서대로 결과값을 출력해야한다는 것이다.
• "번호판 별"로 입출차를 구분해야했기 때문에, dictionary를 이용해서 풀겠다고 생각했다.

 

1.  필요한 변수  및 dictionary 정의

• records에서 입력받은 하나씩 하나씩을 for문을 통해 돌리면서 dictionary에 넣어준다.
• temp에는 record를 띄어쓰기 기준으로 모두 짤라서 넣는다. 
  • 위에 사진에서 입출력 맨 첫번째 records를 예시로 들면, ["05:34", "5961", "IN"] 이렇게 저장될 것이다.
• recordDic에는 key 값은 int로, value는 int 배열 형식으로 받을 것이다.
  • [차량번호: [입차시간, 출차시간, 입차시간]] 이렇게 할 예정이다.
  • if 문을 통해서, carName의 유무를 확인한 뒤, 없다면 key:value를 넣어주고, 있다면 value.append() 해주는 방식으로 값을 넣어줄 수 있다.

let basicTime = fees[0]
let basicFee = fees[1]
let perTime = fees[2]  // 단위 시간
let perFee = fees[3]  //단위요금
var result: [[Int]] = []
var recordDic: [Int: [Int]] = [:]

for record in records {
    let temp = record.split(separator: " ").map{ String($0) }
    let carName = Int(temp[1])!

    if recordDic[carName] == nil {
        recordDic[carName] = [calculateTime(time: temp[0])]
    } else {
        recordDic[carName]!.append(calculateTime(time: temp[0]))
    }
}

 

2. recordDic 에서 출차시간 - 입차시간 계산해주기

• 여기서 제시된 금액은 전부 '분'단위로 계산된다. 
• 따라서 입출차시간을 모두 분으로 바꾸어주는 함수를 만들었다.
  • time에는 string형태로 값을 받는다. "23:59" 이런 형태
  • 해당 시간을 ":" 를 기준으로 짤라서, 23과 59로 나눈다.
  • 23은 hour이기 떄문에 60을 곱해주고, 59는 그냥 분이기 때문에 놔둔다.
  • 해당 두 값을 더해주면, 23:59이 분으로 표현된다.

func calculateTime(time: String) -> Int {
    let tempTime = time.split(separator: ":").map{Int(String($0))!}
    return tempTime[0]*60 + tempTime[1]
}

 

• recordDic에서 출차시간 - 입차시간을 계산해주기 위해서 for문을 돌린다.
• 처음 for문을 돌리기 전에, value의 갯수를 세어서 out시간이 존재하지 않는 것들에게는 "23:59"를 추가해주는 작업을 거친다.
• while문 설명
  • nowIdx를 1로 설정해주고, 2씩 더해주면서 출차시간 - 입차시간 을 계산해준다. 
  • 2를 더하는 이유는, 오름차순으로 정렬되어있기 때문에 인덱스 번호 1, 3, 5,,, 위치에 있는 값들이 출차시간일 것이다. 그래서 2씩 더해주면서, 한칸 앞에있는 입차시간을 빼준다.
• while 문이 끝나면 result에 자동차 번호판을 가르키는 key와 계산되어 나온 주차시간(resultTotalTime)을 [result]에 넣어준다.
  • 2차원 배열로 한 이유는, 나중에 최종 결과를 낼 떄 "차량 번호판" 기준으로 계산해야하기 때문에, 구분선이 필요하기 때문이다.

for (key, value) in recordDic {
    var resultTotalTime = 0
    var nowIdx = 1

    var tempValue: [Int] = value
    // out이 없는 경우 23시 59분 추가
    if value.count % 2 != 0 {
        tempValue.append(calculateTime(time: "23:59"))
    }

    // 2, 4, 6,,, 번째에서 앞꺼 빼기
    while nowIdx < tempValue.count {
        let calTime = tempValue[nowIdx] - tempValue[nowIdx-1]
        resultTotalTime += calTime
        nowIdx += 2
    }
    result.append([key, resultTotalTime])
}

 

3. 최종 주차비 계산

• 최종 주차비를 계산하는 함수를 만들었다.
• total 타임이 기본 제공하는 주차시간보다 크다면, 차를 구하고, 아니라면 계산할 주차시간은 0이된다.
• basicFee는 필수적으로 내야하는 금액이기 때문에, result에 계싼된 금액들을 더해주는 형식으로 구성했다.

func calculateResult(basicFee: Int, basicTime: Int, perFee: Int, perTime: Int,arr: [Int]) -> Int {
    var total = arr[1]
    var result = basicFee
    
    // 기본 시간을 뺀 주차요금을 낼 시간 구하기
    if total > basicTime {
        total = total - basicTime
    } else {
        total = 0
    }
    
    if total > 0 {
        if total % perTime != 0 {
            result += Int((total / perTime + 1) * perFee)
        } else {
            result += Int(total / perTime * perFee)
        }
    }
    return result
}

 

• 최종적으로 구성된 result 는 다음과 같다. 
  • [[차량번호, 금액], [차량번호, 금액]....]
• 이 리스트를, 차량번호를 기준으로 정렬한 후, 금액들을 출력해야한다.
  • 차량번호, 그니까 인덱스 0번을 기준으로 정렬하는 코드는 인터넷을 참고해서 풀었다. (알아두면 좋을듯!)

let realResult = result.sorted(by: {$0[0] < $1[0]})

var returnValue: [Int] = []
for i in 0..<realResult.count {
    returnValue.append(calculateResult(basicFee: basicFee, basicTime: basicTime, perFee: perFee, perTime: perTime, arr: realResult[i]))
}

 

 

최종코드

https://github.com/deslog/Algorithm/blob/main/Algorithm/Programmers/kakao_%EC%A3%BC%EC%B0%A8%20%EC%9A%94%EA%B8%88%20%EA%B3%84%EC%82%B0/main.swift

GitHub - deslog/Algorithm

func solution(_ fees:[Int], _ records:[String]) -> [Int] {
    let basicTime = fees[0]
    let basicFee = fees[1]
    let perTime = fees[2]  // 단위 시간
    let perFee = fees[3]  //단위요금
    var result: [[Int]] = []
    var recordDic: [Int: [Int]] = [:]
    
    for record in records {
        let temp = record.split(separator: " ").map{ String($0) }
        let carName = Int(temp[1])!
        
        if recordDic[carName] == nil {
            recordDic[carName] = [calculateTime(time: temp[0])]
        } else {
            recordDic[carName]!.append(calculateTime(time: temp[0]))
        }
    }
    
    
    for (key, value) in recordDic {
        var resultTotalTime = 0
        var nowIdx = 1
        
        var tempValue: [Int] = value
        // out이 없는 경우 23시 59분 추가
        if value.count % 2 != 0 {
            tempValue.append(calculateTime(time: "23:59"))
        }
        
        // 2, 4, 6,,, 번째에서 앞꺼 빼기
        while nowIdx < tempValue.count {
            let calTime = tempValue[nowIdx] - tempValue[nowIdx-1]
            resultTotalTime += calTime
            nowIdx += 2
        }
        result.append([key, resultTotalTime])
    }
    
    let realResult = result.sorted(by: {$0[0] < $1[0]})
    
    var returnValue: [Int] = []
    for i in 0..<realResult.count {
        returnValue.append(calculateResult(basicFee: basicFee, basicTime: basicTime, perFee: perFee, perTime: perTime, arr: realResult[i]))
    }
    return returnValue
}
    
func calculateTime(time: String) -> Int {
    let tempTime = time.split(separator: ":").map{Int(String($0))!}
    return tempTime[0]*60 + tempTime[1]
}

func calculateResult(basicFee: Int, basicTime: Int, perFee: Int, perTime: Int,arr: [Int]) -> Int {
    var total = arr[1]
    var result = basicFee
    
    // 기본 시간을 뺀 주차요금을 낼 시간 구하기
    if total > basicTime {
        total = total - basicTime
    } else {
        total = 0
    }
    
    if total > 0 {
        if total % perTime != 0 {
            result += Int((total / perTime + 1) * perFee)
        } else {
            result += Int(total / perTime * perFee)
        }
    }
    return result
}