프로그래머스 LV2 n^2배열 자르기

def solution(n, left, right):
    answer = [];
    for i in range(left,right+1) :
        a = i // n;
        b = i % n;
        answer.append(max(a,b)+1);
    return answer

풀이 및 회고

처음엔 2차원 배열을 사용해야하나 생각이 나질 않아서 구글링으로 풀이를 찾아 풀었다.

배열을 펼쳤을때 같은 규칙을 가지는것은 알고 있었는데 풀이로 해결하는 방법을 잘 몰랐다..