n^2 배열 만들기

문제 설명

정수 n, left, right가 주어집니다. 다음 과정을 거쳐서 1차원 배열을 만들고자 합니다. n행 n열 크기의 비어있는 2차원 배열을 만듭니다. i = 1, 2, 3, ..., n에 대해서, 다음 과정을 반복합니다. 1행 1열부터 i행 i열까지의 영역 내의 모든 빈 칸을 숫자 i로 채웁니다. 1행, 2행, ..., n행을 잘라내어 모두 이어붙인 새로운 1차원 배열을 만듭니다. 새로운 1차원 배열을 arr이라 할 때, arr[left], arr[left+1], ..., arr[right]만 남기고 나머지는 지웁니다. 정수 n, left, right가 매개변수로 주어집니다. 주어진 과정대로 만들어진 1차원 배열을 return 하도록 solution 함수를 완성해주세요.

 

풀이

2차원 배열을 만들어 1차원으로 하나씩 읽고 슬라이스 하는 것은 노가다이지만 크게 힘든 것은 아니다.

하지만 수학적인 특징이 있어서 기록해두는 것이 좋을 것 같다.

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

4x4 행렬의 경우, 순차적으로 인덱스 번호가 있다고 가정하자. 이 때, 4로 나눈 나머지와 몫으로 +1 한 값을 나타내면 

4x4 의 인덱스,  ( i % 4 ) + 1 배열 , ( i // 4 + 1) 배열

각 인덱스를 나머지와 몫의 max 값으로 반환 받으면 n^2 배열을 바로 1차원으로 구할 수 있다. 

 

 

출처

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

프로그래머스