[프로그래머스] n^2 배열 자르기 - JS

n^2 배열 자르기

❓ 문제설명

정수 n, left, right가 주어집니다. 다음 과정을 거쳐서 1차원 배열을 만들고자 합니다.

1. n행 n열 크기의 비어있는 2차원 배열을 만듭니다.
2. i = 1, 2, 3, ..., n에 대해서, 다음 과정을 반복합니다.
   - 1행 1열부터 i행 i열까지의 영역 내의 모든 빈 칸을 숫자 i로 채웁니다.
3. 1행, 2행, ..., n행을 잘라내어 모두 이어붙인 새로운 1차원 배열을 만듭니다.
4. 새로운 1차원 배열을 arr이라 할 때, arr[left], arr[left+1], ..., arr[right]만 남기고 나머지는 지웁니다.

정수 n, left, right가 매개변수로 주어집니다. 주어진 과정대로 만들어진 1차원 배열을 return 하도록 solution 함수를 완성해주세요.

🚫 제한조건

1 ≤ n ≤ 10^7

0 ≤ left ≤ right < n^2

right - left < 10^5

✔ 입출력 예

n	left	right	result
3	2	5	[3,2,2,3]
4	7	14	[4,3,3,3,4,4,4,4]

💡 풀이

function solution(n, left, right) {
  // 내풀이
  // answer는 해당 행의 n * n 행렬에서의 위치로 값을 특정할 수 있다
  // [row, col] 위치의 값은 row > col 일 경우 row + 1 이고 그 외에는 col + 1 이다.
  return new Array(right - left + 1).fill(0).map((_, i) => {
    const row = Math.floor((i + left) / n)
    const col = i + left - row * n

    return row > col ? row + 1 : col + 1
  })
}

📝 해설

위 조건에 만족하는 배열에서 [row, col] 좌표의 값은 row + 1 과 col + 1 중 큰 값을 그 값으로 가진다. 또한, 해달 2차원 배열을 조건과 같이 1차원 배열로 만들었을 때 [row, col] 은 [row * n + col] 로 매칭된다. 이러한 특성을 사용하여 미리 left와 right 로 자른 부분배열의 크기의 1차원 배열을 만들고 left 부터 right 까지 반복하며 원본배열에서의 좌표를 찾아 그 값을 계산하여 해결하였다.

📚 참고자료

- https://kimby.tistory.com/48 (Array 메소드)