[Kotlin] 프로그래머스 > 코딩테스트 연습 > 탐욕법(Greedy) > 큰 수 만들기

문제 설명

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

 

풀이

import java.util.*

class Solution {
    fun solution(number: String, k: Int): String {
        val stack = Stack<Char>()
        var del = k
        number.forEach {
            while(stack.isNotEmpty() && stack.peek() < it && del > 0 ) {
                stack.pop()
                del--
            }
            stack.push(it)
        }
        for(i in 1..del) stack.pop()
        return stack.toArray().joinToString("")
    }
}

- 각 숫자의 순서를 바꾸지 않으면서 가장 큰 수를 만들기 위해서는 비교적 앞쪽에 있는 수에서 작은 숫자를 쳐내야 한다. 

- 우선 결과가 될 스택을 만들어둔다. 기본적으로 숫자들을 하나씩 스택에 넣는다. 하지만 넣기전에 스택에 있는 숫자를 하나씩 확인하면서 그 숫자보다 작은 숫자가 스택에 있는 경우 스택에서 뽑는다 = 해당 숫자는 삭제! 

 

---

https://programmers.co.kr/learn/challenges