[프로그래머스] 뒤에 있는 큰 수 찾기 - JavaScript

문제 출처

Lv.2 뒤에 있는 큰 수 찾기 - JavaScript

문제 설명

정수로 이루어진 배열 numbers가 있습니다. 배열 의 각 원소들에 대해 자신보다 뒤에 있는 숫자 중에서 자신보다 크면서 가장 가까이 있는 수를 뒷 큰수라고 합니다.

정수 배열 numbers가 매개변수로 주어질 때, 모든 원소에 대한 뒷 큰수들을 차례로 담은 배열을 return 하도록 solution 함수를 완성해주세요. 단, 뒷 큰수가 존재하지 않는 원소는 -1을 담습니다.

제한사항
  • 4 ≤ numbers의 길이 ≤ 1,000,000
    • 1 ≤ numbers[i] ≤ 1,000,000
예시

입출력 예

numbers return
[2, 3, 3, 5] [3, 5, 5, -1]
[9, 1, 5, 3, 6, 2] [-1, 5, 6, 6, -1, -1]
풀이
function solution(numbers) {
  // 먼저 numbers의 길이 만큼 -1을 채운 배열을 만들어준다.
  const answer = Array.from({ length: numbers.length }, () => -1);

  // 인덱스를 넣어줄 배열
  const idxs = [];

  // numbers의 길이 만큼 for문을 돌려 하나씩 확인한다.
  for (let i = 0; i < numbers.length; i++) {
    // idxs를 하나씩 확인해준다.
    // 조건 - 인덱스가 존재하고 해당 존재하는 인덱스 순서의 숫자가 현재 수 보다 낮을경우
    while (idxs.length && numbers[idxs[idxs.length - 1]] < numbers[i]) {
      // answer의 인덱스 위치에 현재 들어온 수를 넣어준다.
      answer[idxs.pop()] = numbers[i];
    }

    // 현재 인덱스를 넣어준다.
    idxs.push(i);
  }

  // 마지막은 확인할 수 없으므로 미리 셋팅한 -1 그대로다
  return answer;
}
정리

사용된 메서드와 문법에 대해 더 공부하고 싶으신 분은 링크를 클릭해주세요!

Array.from() 메서드 - MDN

피드백은 언제나 환영입니다. 😊