[프로그래머스] Level 1. K번째수 (JS)

문제 설명

배열 array의 i번째 숫자부터 j번째 숫자까지 자르고 정렬했을 때, k번째 수를 구하려고 한다.
예를 들어 array가 [1, 5, 2, 6, 3, 7, 4], i = 2, j = 5, k = 3 이라면
    1. array의 2번째부터 5번째까지 자르면 [5, 2, 6, 3]
    2. 1에서 나온 배열을 정렬하면 [2, 3, 5, 6]
    3. 2에서 나온 배열의 3번째 숫자는 5이다.

 

입출력 예시

array	commands	return
[1, 5, 2, 6, 3, 7, 4]	[[2, 5, 3], [4, 4, 1], [1, 7, 3]]	[5, 6, 3]

 

문제 풀이

function solution(array, commands) {
  let answer = [];
  
  // commands의 length만큼 순회하면서 i, j, k를 확인한다.
  for(let i = 0; i < commands.length; i++) {
  
    // newArr는 기존 array에서 i번째에서 j번째까지 자른 배열이다.
    let newArr = array.slice(commands[i][0] - 1, commands[i][1])
    
    // newArr에서 k번째에 있는 수를 구해야하기 때문에 먼저 sort 메소드로 정렬을 해준다.
    newArr.sort(a, b) => a - b)
    
    // 정렬이 됐다면 k번째의 수를 빈 배열(answer)에 차례대로 넣어준다.
    answer.push(newArr[commands[i][2] - 1])
  }
  
  return answer;
}