n^2 배열 자르기

yejichoi·2023년 8월 7일
0

알고리즘 스터디

목록 보기
94/153
post-thumbnail

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 함수를 완성해주세요.

제한사항

1 ≤ n ≤ 10^7
0 ≤ left ≤ right < n2
right - left < 10^5

입출력 예

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

풀이

처음에는 이중 포문으로 이차원 배열 만들어서 접근
근데 주어진 숫자가 너무 커서 결국 못품

다른 사람 아이디어

- 인덱스의 몫과 나머지로 n x n 배열의 행과 열을 알 수 있습니다. 그 중 큰 값에 1을 더하면 배열의 값

def solution(n, left, right):
   answer = []
   for i in range(left, right+1):
       v = max(i//n, i%n) + 1
       # x=i//n, y=i%n | x, y 중에 큰 수를 구해 +1을 해서 값으로 넣어줌
       print(i//n, i%n,v) # 몫, 나머지 
       answer.append(v)
   return answer
   ```

0개의 댓글