[알고리즘] n^2 배열 자르기 (프로그래머스 2레벨)

 

문제

https://www.acmicpc.net/problem/16967

 

📌 알고리즘 풀이

class Solution {
      public int[] solution(int n, long left, long right) {

        int[] arr = new int[(int)(right-left+1)];
        
        for(long i=left; i<=right; i++) {
            int quotient = (int)(i/n);  // 몇 번 째 row인지 (0부터 시작)          
            int rest = (int) (i%n);  // 해당 row 에서 column 번째 열인지 (0부터 시작)
            
            int idx = (int)(i-left);
            
            if(quotient>=rest) {
                arr[idx] = quotient+1;
                continue;
            }
            arr[idx] = rest+1;
        }
        return arr;
    }
}