프로그래머스 [Lv. 2] 뒤에 있는 큰 수 찾기 {언어 : JavaScript} [꼭 다시 보기]

문제 링크

https://school.programmers.co.kr/learn/courses/30/lessons/154539?language=javascript

프로그래머스

정답 코드

function solution(numbers) {
    const n = numbers.length;
    const answer = Array.from({length: n}, () => -1);
    const stack = [];
    for (let i = 0; i < n; i++) {
        while (stack.length) {
            if (numbers[stack.at(-1)] >= numbers[i]) {
                break;
            } else {
                answer[stack.pop()] = numbers[i];
            }
        }
        stack.push(i);
    }
    return answer;
}

풀이 방법

1. stack에 위치값 i를 넣기 전에 stack이 비었는지 확인한다.
2. stack에 값이 들어 있다면 stack의 마지막 위치값 = stack.at(-1)의 위치값에 해당하는 값과 현재 i의 위치값에 해당하는 값을 비교한다.
   1. numbers[stack.at(-1)]이 numbers[i]보다 크거나 같은 경우 반복문을 종료시킨다.
   2. 반대인 경우 numbers[stack.at(-1)]보다 크다는 말은 뒤에 있는 큰수라는 말이므로 answer[stack.pop()]에 numbers[i]값으로 갱신한다.
3. 반복문이 종료되면 stack에 현재의 위치값 i를 넣는다.
4. 이렇게 마지막 인덱스까지 반복하고 answer를 출력하면 끝!

느낀점

• 스택을 이렇게 사용하는 방법이 있다는 것을 다시금 깨달았다.
• 다시 풀어 보는 문제였는데 다음에 또 다시 풀어 보면 좋을 것 같다.