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

n^2 배열 자르기

문제링크

풀이

처음에는 matrix를 2차원 배열로 생성하여 값을 저장후 answer로 값을 추출했는데
메모리 초과 오류가 났다.
이후 규칙으로 수식으로 바로 값을 얻을 수 있도록 고쳐 문제를 해결했다.

코드

class Solution {
    public int[] solution(int n, long left, long right) {
        int[] answer = new int[(int) (right - left + 1L)];
        
        for (int i = 0; i < answer.length; i++) {
            int x = (int)((left + i)/n);
            int y = (int)((left + i)%n);
            answer[i] = Math.max(x, y) + 1;
        }
        
        return answer;
    }
}