[Programmers] n^2 배열자르기

문제 설명

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

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

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

제한사항

• 1 ≤ n ≤ 107
• 0 ≤ left ≤ right < n2
• right - left < 105

입출력 예

|n |left| right| result|
|—|—|—|—|
|3 |2| 5| [3,2,2,3]|
|4| 7| 14| [4,3,3,3,4,4,4,4]|

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

    for i in range(left, right+1):
        answer.append(max(divmod(i, n)) + 1)
    return answer

문제풀이

맨처음에는 단순하게 이차원배열로 이중 for문으로 하나하나 방문하면서 해당 행,열에 맞는 값을 넣어 주었는데, 이러니까 자꾸 시간초과가 떠서 무조건 for문 하나로 풀려고 규칙을 찾아내서 풀었다.

위와 같은 방법으로 행렬을 채우게 되면, 행의 값과 열의 값중에 더 큰 값을 갖는 값 + 1 이 해당 위치의 값이므로, 이차원배열을 divmod()로 몫과 나머지로 받아 그 중 더 큰 값에 +1 을 해서 행렬에 넣어주면된다.