본문 바로가기

# 02/Swift - CTP

[Swift] 코딩테스트 연습! Lv2. 큰 수 만들기

반응형
/* 큰 수 만들기

 - 어떤 숫자에서 k개의 수를 제거했을 때 얻을 수 있는 가장 큰 숫자를 구하려 합니다.
 
 예를 들어, 숫자 1924에서 수 두 개를 제거하면 [19, 12, 14, 92, 94, 24] 를 만들 수 있습니다. 이 중 가장 큰 숫자는 94 입니다.

 문자열 형식으로 숫자 number와 제거할 수의 개수 k가 solution 함수의 매개변수로 주어집니다. number에서 k 개의 수를 제거했을 때 만들 수 있는 수 중 가장 큰 숫자를 문자열 형태로 return 하도록 solution 함수를 완성하세요.

 제한 조건
 number는 2자리 이상, 1,000,000자리 이하인 숫자입니다.
 k는 1 이상 number의 자릿수 미만인 자연수입니다.
*/
func solution3(_ number:String, _ k:Int) -> String {
    var numberList:[Character] = Array(number)
    var count = k
    var index = 0
    var result:[Character] = []
    while count > 0 {
        if index+count == numberList.count {
            return String(result)
        }
        var maxValue:Character = numberList[index]
        var maxIndex:Int = index
        for i in index...index+count {
            if numberList[i] == "9" {
                maxValue = "9"
                maxIndex = i
                break
            } else if maxValue < numberList[i] {
                maxValue = numberList[i]
                maxIndex = i
            }
        }
        count -= maxIndex-index
        index = maxIndex+1
        result.append(maxValue)
    }
    if index < number.count-1 {
        result += numberList[index...number.count-1]
    }
    return String(result)
}
반응형