프로그래머스 큰 수 만들기 Swift

문제 설명

어떤 숫자에서 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

풀이: 우선 number를 map으로 나눠준다. 이렇게 하는 이유는 글자를 배열처럼 다룰려면 index값을 정해줘야하기 때문에 귀찮아진다.

그리곤 수들을 담을 배열을 만들어주고 삭제한 수를 세는 count라는 변수도 만들어준다.

그리곤 for문을 number의 길이에 맞게 돌리고 for문 안에는 while문으로 3가지 조건을 달아준다.

1.삭제한수(count)가 k를 넘지 않을 것

2.stack이 비어있지 않을 것

3.스택의 가장 최근의 값이 그 다음으로 들어오는 값보다 크지 않을 것

아렇게 3가지 조건을 충족한다면 stack의 가장 최근의 값을 지워준다.

왜냐하면 그 다음으로 들어오는 수가 가장 최근의 값보다 크다면 

삭제되어야 하기 때문이다. 그리곤 count를 1 더해준다.

while문을 빠져나와서 count가 k와 같거나 더 커졌다면 

stack에 map의 지금부터 끝까지 넣어준다, stack.append(map[i]..). (contentsOf:a...b -> a부터 b까지 배열에 넣는다.)

만약 아직 k를 넘지 않았다면 map[i]만 넣어주도록 한다.

그리고 마지막엔 stack을 모두 합친 글자 stack.joined에서 우리가 구해야 하는 길이인 number의 길이에서 k를 뺀 값만

String형태로 반환해준다.

import Foundation
 
func solution(_ number:String, _ k:Int) -> String {
    let map = number.map{String($0)}
    var stack = [String]()
    var count = 0
    for i in 0..<number.count{
        while count < k && !stack.isEmpty && stack.last! < map[i] {
            stack.removeLast()
            count += 1
        }
            if count >= k {
                stack.append(contentsOf: map[i...])
                break
            }else{
                stack.append(map[i])
        }
    }
    return String(stack.joined().prefix(number.count-k))
}
 

계속 8번과 10번에 시간초과가 났었는데 contentsOf와 prefix로 number.count-k로 길이를 정해주니깐 시간초과가 해결되었다.