[Programmers] Level 2. n^2 배열 자르기

🔗 Problem Link

Level 2. n^2 배열 자르기

❔Thinking

• n^2배열을 만들고, 이를 left에서 right까지 자른 결과를 출력한다.

💻Solution

def solution(n, left, right):
    start, end = left//n, right//n
    left, right = left-(start*n), right-(start*n)
    result = []
    for i in range(start, end+1):
        result += [i+1 for _ in range(i+1)] + [j for j in range(i+2, n+1)]
    return result[left:right+1]

🗝️keypoint

1. n^2 배열을 for문으로 만들면 시간초과가 발생한다.
2. 주어진 배열에 대한 규칙을 찾고, left~right까지만 구해 출력해야 한다.