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

최초 코드 - 시간 초과

def solution(n, left, right):
    answer = [[] for _ in range(n)]
    
    for i in range(n):
        for j in range(i + 1):
            answer[i].append(i + 1)
        for k in range(i + 1, n):
            answer[i].append(k + 1)

    result = []
    for i in answer:
        result = result + i
    
    return result[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