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

https://school.programmers.co.kr/learn/courses/30/lessons/87390#

프로그래머스

---

처음에는 n^2 배열을 만든 뒤, 리스트 슬라이싱을 이용해서 값을 추출했었는데

그렇게 푸니까 시간초과가 났다.

 

그래서 배열상 인덱스를 이용해서 다시 풀었다.

 

python

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

    for i in range(left, right+1):
        row = i // n
        col = i % n
        if row > col:
            answer.append(row+1)
        else:
            answer.append(col+1)
    return answer