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

김상현·2022년 9월 20일
0

알고리즘

목록 보기
189/301
post-thumbnail

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

📍 문제 설명

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

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

📍 입출력 예

nleftrightresult
325[3,2,2,3]
4714[4,3,3,3,4,4,4,4]

📍 문제 풀이

  • n 이 3인 경우 만들어지는 1차원 배열은 아래와 같습니다.
idx012345678
value123223333
  • 1차원 배열의 인덱스 값의 몫과 나머지를 구하면 아래와 같습니다.
idx012345678
value123223333
quotient000111222
remainder012012012
  • value 값과 몫(quotient) 그리고 나머지(remainder) 와의 관계는 value = max(quotient, remainder) + 1 와 같다.
  • left 가 2이고, right 가 5인 경우를 살펴보면 아래와 같다.
idx2345
value3223
quotient0111
remainder2012
[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)]
profile
목적 있는 글쓰기

0개의 댓글