[프로그래머스] 큰 수 만들기 | JavaScript

예구·2023년 9월 16일

JavaScript Lv.2 알고리즘 프로그래머스

Algorithm

목록 보기

40/47

문제 출처

1. 문제

문제 설명

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

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

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

제한 조건

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

입출력 예

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

2. 풀이

처음에는 아래와 같은 방식으로 풀어보려고 했다.

number.slice(0, number.length-k)의 숫자 중 가장 큰 수 찾기
찾은 숫자를 startIdx라고 할 때, startIdx의 다음 index부터 index+1보다 작다면, 해당 index의 값을 없애기
이 과정에서 k에 없앤 숫자들을 빼주기

하지만 이렇게 풀었더니 입출력 예제는 통과하였지만 제출하였을 때 문제를 틀렸다. 따라서 다른 방법을 찾아야 했다.

찾아보니 stack을 이용해서 푸는 방법이 있었다.

number를 회문을 돌며, stack의 마지막 숫자와 현재 숫자를 비교하여 현재 숫자가 더 크다면 stack의 마지막 숫자를 pop()
- stack의 마지막 숫자가 현재 숫자보다 클 때까지 최대 k번 반복함
number의 모든 숫자를 비교한 후, k가 0보다 크다면 stack의 남은 k만큼 뒤에서 제거함

전체 풀이는 아래와 같다.

function solution(number, k) {
  let stack = [];

  for (let i = 0; i < number.length; i++) {
    const element = numbeㅋr[i];
    // stack 마지막 숫자와 현재 숫자를 비교하여 현재 숫자가 더 크면 stack의 마지막 숫자 pop
    while (k > 0 && stack[stack.length - 1] < element) {
      stack.pop();
      k--;
    }
    stack.push(element);
  }

  // 모든 숫자 비교가 끝난 후, k가 0보다 크다면 남은 k만큼 뒤에서 제거
  stack.splice(stack.length - k, k);
  return stack.join("");
}