[프로그래머스 LEVEL2] n^2 배열 자르기 / Javascript

문제 설명

정수 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]

풀이 과정

처음에는 문제에 나와있는 조건대로 2차원 배열을 만들고 배열에 값을 채워넣고

그걸 또 1차원 배열로 변환하고 그 다음에 정해진 범위만큼 잘라서 리턴하도록 했다.

그랬더니 거의 모든 테케에서 터지고 말았다..

왜냐..? n의 최댓값이 10^7이기 때문이다...^^

만일 n이 10^7이라면 arr[10^7][10^7] 배열을 만들고 여기서 이중 반복문을 돌리는

과정을 거쳐야 하는건데 안 터지는게 더 이상하다..

그래서 효율성을 위해 이차원 배열을 만들지 않고 처음부터 arr[left]~arr[right]까지만

구하는 방법을 생각했다. 

 

먼저, 2차원 배열의 행과 열, 원소값, 1차원 배열의 인덱스간에 규칙성을 찾기 위해 이 값들을 좌르륵 나열해봤다.

나열해놓고 보니 정말 쉽게 규칙성을 찾을 수 있었다.

1. row = k / n
2. column = k % n
3. row가 column보다 같거나 크면 num = row + 1
4. row가 column보다 작으면 num = column + 1

어때요..참 쉽쥬..?

이렇게 찾은 규칙을 코드로 구현했더니 쉽게 통과할 수 있었다.

function solution(n, left, right) {
    const answer = [];
    for(let k = left; k <= right; k++) {
        const row = Math.floor(k / n);
        const column = k % n;
        if(row >= column) {
            answer.push(row + 1);
        }
        else {
            answer.push(column + 1);
        }
    }
    return answer;
}