n^2 배열 자르기

https://school.programmers.co.kr/learn/courses/30/lessons/87390

프로그래머스

 

 

2차원 배열을 row 단위로 잘라 이어 붙여서 1차원 배열로 만들고, 주어진 left, right 구간 배열을 리턴해야 한다.

 

2차원 배열을 만들기에는 n의 단위가 10^7 이므로 무리.

 

문제 샘플로 주어지는 예시를 잘 살펴보면

 

행과 열에 따라 어떤 숫자를 넣어야 하는지 알 수 있다.

 

class Solution {
    public int[] solution(int n, long left, long right) {
        
        int [] answer = new int[((int) (right - left)) + 1];
        
        int index = 0;
        for (long i = left; i <= right; i++) {
            
            int row = (int) (i / n);
            int col = (int) (i % n);
            
            int num = 0;
            if (col <= row) {
                num = row + 1;
            } else {
                num = col + 1;
            }
            
            answer[index++] = num;

        }
        
        return answer;
    }
}