[백준 17298번] 오큰수 - 자바스크립트(nodejs)

https://www.acmicpc.net/problem/17298

17298번: 오큰수

const fs = require('fs');
let input = fs.readFileSync('dev/stdin').toString().trim().split('\n');
input.shift();
input = input[0].split(' ').map((v) => +v);

let arr = [];
let answer = new Array(input.length).fill(-1);

for (let i = 0; i < input.length; i++) {
  while (arr.length && input[arr[arr.length - 1]] < input[i]) {
    answer[arr.pop()] = input[i];
  }
  arr.push(i);
}
console.log(answer.join(' '));

우선 input 길이만큼 -1을 채운 배열을 생성한다.

 

그 후 단축평가를 통해 arr배열의 길이가 0이 아닌 값에 대해 input[i]값과 input[arr[arr.length-1]]의 값을 비교한다.

즉 왼쪽 값과 비교해서 자신보다 큰 값을 찾는 것이다. 

 

3,4,6,7을 예로 들어보자.

input = [3,5,2,7]

answer = [-1, -1, -1, -1]

arr = [ ]

 

i = 0       // 위 while문 0번 순회

answer = [-1, -1, -1, -1]

arr = [0]

 

i = 1        // 위 while문 1번 순회

answer = [ 5, -1, -1, -1 ]
arr = [ 1 ]

 

i = 2      // 위 while문 0번 순회

answer = [ 5, -1, -1, -1 ]
arr = [ 1 , 2]

 

i = 3     // 위 while문 2번 순회

answer = [ 5, 7, 7, -1 ]
arr = [3]

 

코드와 함께 보면 쉽게 이해할 수 있다.