[알고리즘] Programmers n^2 배열 자르기 #Python

[Programmers] n^2 배열 자르기 바로가기

📍 문제 설명

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

1. n 행 n 열 크기의 비어있는 2차원 배열을 만듭니다.
2. i = 1, 2, 3, ..., n 에 대해서, 다음 과정을 반복합니다.
   • 1행 1열부터 i 행 i 열까지의 영역 내의 모든 빈 칸을 숫자 i 로 채웁니다.
3. 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]

---

📍 문제 풀이

• n 이 3인 경우 만들어지는 1차원 배열은 아래와 같습니다.

idx	0	1	2	3	4	5	6	7	8
value	1	2	3	2	2	3	3	3	3

• 1차원 배열의 인덱스 값의 몫과 나머지를 구하면 아래와 같습니다.

idx	0	1	2	3	4	5	6	7	8
value	1	2	3	2	2	3	3	3	3
quotient	0	0	0	1	1	1	2	2	2
remainder	0	1	2	0	1	2	0	1	2

• value 값과 몫(quotient) 그리고 나머지(remainder) 와의 관계는 value = max(quotient, remainder) + 1 와 같다.
• left 가 2이고, right 가 5인 경우를 살펴보면 아래와 같다.

idx	2	3	4	5
value	3	2	2	3
quotient	0	1	1	1
remainder	2	0	1	2

[idx = 2] value = max(0, 2) + 1 = 3
[idx = 3] value = max(1, 0) + 1 = 2
[idx = 4] value = max(1, 1) + 1 = 2
[idx = 5] value = max(1, 2) + 1 = 3

✍ 코드

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