문제 설명

어떤 숫자에서 k개의 수를 제거했을 때 얻을 수 있는 가장 큰 숫자를 구하려 합니다.

예를 들어, 숫자 1924에서 수 두 개를 제거하면 [19, 12, 14, 92, 94, 24] 를 만들 수 있습니다. 이 중 가장 큰 숫자는 94 입니다.

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

제한 사항

  • number는 1자리 이상, 1,000,000자리 이하인 숫자입니다.
  • k는 1 이상 number의 자릿수 미만인 자연수입니다.

입출력 예

number k return
“1924” 2 “94”
“1231234” 3 “3234”
“4177252841” 4 “775841”

입출력 예 설명

문제에 나온 예와 같습니다.

문제 풀이

  • 일정 배열에서 가장 큰수를 max()를 이용해서 풀었는데 시간초과가 발생하였다. 
    var checkingIndex = numberArr.firstIndex(numberArr[0...count].max())

  • 그 이유는 가장 큰 자릿수에서부터 작은 자릿수로 순회하는데 가장 큰 수인 9가 중간에 나오면 더 이상 뒤를 순회 하지 않아도 되기 때문이다.

  • 그래서 9가 나오면 순회를 멈추도록 설정 해주었더니 시간초과를 해결할 수 있었다.

import Foundation

func solution(_ number:String, _ k:Int) -> String {
    
    var numberArr: [Int] = number.compactMap { Int(String($0)) }
    var count = k
    var answer = ""
    
    while count > 0 {

        if numberArr.count == count { break }

        var checkingIndex = 0

        for i in 0...count {
            if numberArr[i] == 9 {
                checkingIndex = i
                break
            }
            if numberArr[checkingIndex] < numberArr[i] { checkingIndex = i }
            
        }
        
        answer += String(numberArr[checkingIndex])
        count -= checkingIndex
        numberArr.removeFirst(checkingIndex+1)
    }
    
    if count == 0 && !numberArr.isEmpty {
        answer += numberArr.reduce("") { $0 + String($1) }
    }
    
    return answer
}

참고 : https://programmers.co.kr/learn/courses/30/lessons/42883