[프로그래머스][JAVA]Lv. 2 - n^2 배열 자르기

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

프로그래머스

n^2인 배열을 잘라 1차원 배열로 만들고 left와 right사이를 잘라서 리턴하는 문제

import java.util.ArrayList;
import java.util.Arrays;
class Solution {
    public int[] solution(int n, long left, long right) {
        int[] arr = new int[(int) (right-left+1)];
        long s = left / n + 1;
        long s2 = left % n + 1;
        for (long i = 0; i < arr.length; i++) {
            if (s2 >= n+1) {
                s2 = s2-n;
                s++;
            }
            if (s >= s2) {
                arr[(int) i] = (int) s;
            } else {
                arr[(int) i] = (int) s2;
            }
            s2++;
        }

        return arr;
    }
}