[프로그래머스] 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⁷
• 0 ≤ left ≤ right < n²
• right - left < 10⁵

입출력 예

수도 코드

1. 매트릭스를 만들어서 행, 열을 기준으로 반복문을 돌며 해당 인덱스 + 1의 값을 찍는다
2. flat(2)함수를 통해 2차원배열을 풀어주고 left에서 right까지 값을 리턴한다.
   // 런타임 아웃 실패

1. left를 n으로 나눠줌으로써 해당 row값을 얻는다.
2. row값이 col값보다 크면 해당 row에 해당하는 숫자를 찍어준다
3. 그이상이면 row와 col의 차이를 row에 더해 숫자를 찍어준다.
   // 테스트 100%

Solution

function solution(n, left, right) {
    var answer = [];
    let len = right + 1 - left
    while (answer.length < len) {
        if (left < n) {
            answer.push(left + 1)
            left++
        } else {
            let row = parseInt(left / n)
            let col = left % n
            if (row >= col) {
                answer.push(row + 1)
                left++
            } else {
                let op = col - row
                answer.push(row + 1 + op)
                left++
            }
        }
    }
    return answer
}

GItHub repo