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

코딩테스트 연습 - n^2 배열 자르기

정수 n, left, right가 주어집니다. 다음 과정을 거쳐서 1차원 배열을 만들고자 합니다. n행 n열 크기의 비어있는 2차원 배열을 만듭니다. i = 1, 2, 3, ..., n에 대해서, 다음 과정을 반복합니다. 1행 1열부

programmers.co.kr

left와 right를 나열해서 쓰다 보면 left 값과 right 값으로 위치를 찾을 수 있다.

이를 통해 풀이하면 된다.

나의 풀이

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

With Data