어떤 숫자에서 k개의 수를 제거했을 때 얻을 수 있는 가장 큰 숫자를 구하려 합니다.
예를 들어, 숫자 1924에서 수 두 개를 제거하면 [19, 12, 14, 92, 94, 24] 를 만들 수 있습니다. 이 중 가장 큰 숫자는 94 입니다.
문자열 형식으로 숫자 number와 제거할 수의 개수 k가 solution 함수의 매개변수로 주어집니다. number에서 k 개의 수를 제거했을 때 만들 수 있는 수 중 가장 큰 숫자를 문자열 형태로 return 하도록 solution 함수를 완성하세요.
const solution = (number, k) => {
let arr = [];
for(let i = 0; i < number.length; i++) {
const num = number[i];
while(k > 0 && arr.at(-1) < num) {
arr.pop();
k--;
}
arr.push(num);
}
arr.splice(arr.length - k, k);
return arr.join('')
}
문제 제한사항을 보고 반복문을 막 쓰면 시간초과가 날 것 이라는 생각이 들었다.
처음 문제를 보았을때, 정렬해서 풀면 되는 줄 알았으나, 다시보니 문자열의 순서 그대로 유지해야되는 것이였다.
코드를 설명하자면, 우선 스택을 하나 선언한다.
number
문자열의 숫자들을 하나씩 넣으면서 스택의 top
값이 num
값 보다 작으면 빼내고 k 를 감소시킨다.
스택에 k 개 만큼 필요하지 않은 요소가 존재할 수 있어서 arr.length - k
값 부터 k
만큼 제거한다.
이런 유형의 문제가 자주 출시되는것 같다.
제한 사항은 1,000,000 까지고 스택을 사용하여 크기 비교하는 문제 유형을 몇개 더 풀어봐야 되겠다.