프로그래머스 - n^2 배열 자르기

---

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

코딩테스트 연습 - 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 ≤ 10^7
• 0 ≤ left ≤ right < n^2
• right - left < 10^5

입출력 예

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):
    arr=[[i for i in range(1,n+1)]]*n#제일 첫번째 행
    ans=[]
    for i in range(left//n,right//n+1):# 구해야 하는 크기만큼만 규칙에 맞게 행을 생성
        ans+=[i+1]*(i)+ arr[0][i:]
    return ans[left%n:left%n+(right-left)+1]#슬라이싱하여 리턴

 

n^2의 배열 리스트는 두번째 행부터 [행의 인덱스*행의 길이] +[행의 인덱스: 끝]으로 진행된다. 예를 들어 n=3 일 경우 1행은 [1,2,3] 이지만 그 다음 두번째 행부터는 [2,2,3]이고, 마지막 행은 [3,3,3]이다. 

이렇게 찾아낸 규칙에 따라 구해야 하는 크기만큼만 행을 생성하여 정답 리스트에 추가한 후 구해야 하는 크기만큼 정답 리스트를 다시 슬라이싱하여 리턴시켰다.

정답이긴 한데, 용량도 엄청 쓰는데다 만약 배열의 크기가 더 컸을 경우에는 시간초과 100퍼 났을 것 같다.

 

다른 사람이 짠 코드 중 대단하다 싶었던 것

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

    return answer

어떻게 돌아가는지 이해는 가는데;;;; 왜 이렇게 진행이 되는지 잘 모르겠다.