Vue/JavaScript 알고리즘

[JS 프로그래머스] 큰수구하기

힛해 2024. 1. 18. 15:31
728x90

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

 

프로그래머스

코드 중심의 개발자 채용. 스택 기반의 포지션 매칭. 프로그래머스의 개발자 맞춤형 프로필을 등록하고, 나와 기술 궁합이 잘 맞는 기업들을 매칭 받으세요.

programmers.co.kr

문제 설명

 

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

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

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

function solution(number, k) {
    
    let stack = [];
    let removeCount = 0;
    for(let i =0 ; i < number.length; i++){
        while(removeCount < k && stack.length && number[i] > stack[stack.length-1]){
              stack.pop();
            removeCount++;            
              }
        stack.push(number[i]);
    }
    answer = stack.join('');
   return answer;
}

 

문제 해결 방식

이 문제는 숫자를 접근할때 그 전의 숫자가 본인보다 작으면 지우면 된다.

입력값이 number = 1924 , k =2 라 했을 때

1 -> 9 1은 지워져야함 9 -> 1 괜찮다 1->4 지워져야한다.

이런 방식의 알고리즘

function solution(number, k) {
    let stack = [];
    let removeCount = 0;
    for(let i =0 ; i < number.length; i++){
        while(removeCount < k && stack.length && number[i] > stack[stack.length-1]){
              stack.pop();
            removeCount++;            
              }
        stack.push(number[i]);
    }

 

removeCount가 전부 할당되지 않고, stack이 비어있지 않고

현재 접근한 숫자가 stack의 끝에 쌓여있는 숫자보다 더 클때

stack에 저장한 숫자를 삭제한다.

이때 while문이기에

stack [ 1, 1 ]  number[i] = 3 일 경우 1과 1 모두 삭제한다