[프로그래머스/파이썬] n^2 배열 자르기

 

 

나의 풀이

def solution(n, left, right):
    array = []
    for i in range(left, right + 1):
        array.append(max(i // n, i % n) + 1)

    return array

 

 

divmod를 이용한 풀이

def solution(n, left, right):
    answer = []

    for i in range(left, right+1):
        q, r = divmod(i, n)
        answer.append(max(q, r) + 1)

    return answer