[JavaScript] 프로그래머 수열과 구간 쿼리 2

문제 설명

정수 배열 arr와 2차원 정수 배열 queries이 주어집니다. queries의 원소는 각각 하나의 query를 나타내며, [s, e, k] 꼴입니다. 각 query마다 순서대로 s ≤ i ≤ e인 모든 i에 대해 k보다 크면서 가장 작은 arr[i]를 찾습니다. 각 쿼리의 순서에 맞게 답을 저장한 배열을 반환하는 solution 함수를 완성해 주세요. 단, 특정 쿼리의 답이 존재하지 않으면 -1을 저장합니다.

 

입출력 예

arr	queries	result
[0, 1, 2, 4, 3]	[[0, 4, 2],[0, 3, 2],[0, 2, 2]]	[3,4,-1]

코드

const solution = (arr, queries)=>{
  let answer = [];
  for (query of queries) {
    const [s, e, k] = query;
    const line = arr.slice(s, e + 1).filter((v) => v > k);
    answer.push(line.length > 0 ? Math.min(...line) : -1);
  }
  return answer;
}

 

 

설명

1) slice를 사용하여 현재 쿼리의 범위에 해당하는 부분 배열을 추출한 후 filter를 사용하여 k보다 큰 값을 필터링합니다.

2) 필터링된 부분이 배열이 비어있지 않으면 최소값을 찾아 반환하고 그렇지 않으면 -1 반환합니다.

3) 현재 쿼리의 결과를 answer 배열에 추가합니다.