[JAVA] Pro n^2 배열 자르기

문제

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

프로그래머스

코드

class Solution {
    public int getNumber(long idx, int n){
        // 나머지 <= 몫
        if(idx%n<=idx/n){
            return (int)(idx/n+1);
        }
        else{
            return (int)(idx%n+1);
        }
    }
    
    public int[] solution(int n, long left, long right) {
        int[] answer = new int[(int)(right-left+1L)];
        
        for(long i=left; i<=right; i++){
            answer[(int)(i-left)]=getNumber(i, n);
        }
        
        return answer;
    }
}