programmers | Lv2. n^2 배열 자르기 [Python]

💡 Python 3

🔗 문제

n^2 배열 자르기 [Link]

💻 코드

시간 초과 코드 !! 으아ㅏㅏ 코드 잘 짜고 싶다!!!!

def solution(n, l, r):
    arr = []
    for i in range(n):
        for j in range(n):
            arr.append(max(i+1, j+1))
    return arr[l:r+1]

💥 다른 사람 코드

서치 해봤다.
위치값에서 n을 나눠준 나머지와 몫 중 큰 값에 1을 더한 값이 해당 위치의 값이었다..

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

같은 방법에 lambda 사용한 코드

solution = lambda n, left, right: list((max(i // n, i % n) + 1 for i in range(left, right + 1)))

참고 자료

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