n^2 배열 자르기

문제 안내

문제 설명

• 정수 n, left, right가 주어집니다. 다음 과정을 거쳐서 1차원 배열을 만들고자 합니다.

1. n행 n열 크기의 비어있는 2차원 배열을 만듭니다.
2. i = 1, 2, 3, ..., n에 대해서, 다음 과정을 반복합니다.
3. 1행 1열부터 i행 i열까지의 영역 내의 모든 빈 칸을 숫자 i로 채웁니다.
4. 1행, 2행, ..., n행을 잘라내어 모두 이어붙인 새로운 1차원 배열을 만듭니다.
   새로운 1차원 배열을 arr이라 할 때, arr[left], arr[left+1], ..., arr[right]만 남기고 나머지는 지웁니다.

• 정수 n, left, right가 매개변수로 주어집니다.
• 주어진 과정대로 만들어진 1차원 배열을 return 하도록 solution 함수를 완성해주세요.

제한사항

입출력 예

풀이

규칙 찾기

	solution(3, 2, 5)에 대해 살펴보자.
    
	1. 3 x 3 사이즈의 2차원 배열은 아래와 같이 생김
         [[1,2,3], [1,2,3], [1,2,3]]
    2. 총 9개의 요소를 좌표로 표현 시,
     	 (0, 0) = 1
         (0, 1) = 2
         (0, 2) = 3
         (1, 0) = 1
         (1, 1) = 2
         (1, 2) = 3
         (2, 0) = 1
         (2, 1) = 2
         (2, 2) = 3
        
    3. 문제 조건대로 해당 배열을 변경 시,
    	 (0, 0) = 1
         (0, 1) = 2
         (0, 2) = 3
         (1, 0) = 2    ←
         (1, 1) = 2
         (1, 2) = 3
         (2, 0) = 3    ←
         (2, 1) = 3    ←
         (2, 2) = 3
    4. 변경된 후 좌표와 요소와의 연관성을 살펴보면 다음과 같음.
         x, y 좌표의 최대값에 1을 더한 값 = 해당 요소의 값

Code

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