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

문제

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

시행착오

• 시간 초과 -> 너무 정직하게 풀었다. 사칙연산을 이용하기
• i//n,i%n 여기까지는 알았는데.. max라고?

def solution(n, left, right):
    arr = [[0] * n for _ in range(n)]
    for i in range(n):
        for j in range(i+1):
            arr[i][j] = i+1
            arr[j][i] = i+1
        
    answer = []
    for i in arr:
        answer += i
    return answer[left:right+1]

풀이

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