728x90
반응형
n^2 배열 자르기
❓ 문제설명
정수 n, left, right가 주어집니다. 다음 과정을 거쳐서 1차원 배열을 만들고자 합니다.
- n행 n열 크기의 비어있는 2차원 배열을 만듭니다.
- i = 1, 2, 3, ..., n에 대해서, 다음 과정을 반복합니다.
- 1행 1열부터 i행 i열까지의 영역 내의 모든 빈 칸을 숫자 i로 채웁니다. - 1행, 2행, ..., n행을 잘라내어 모두 이어붙인 새로운 1차원 배열을 만듭니다.
- 새로운 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 메소드)
728x90
반응형