[프로그래머스][Lv. 2] n^2 배열 자르기

junhyeong04·2023년 9월 25일

codingTestPython

목록 보기
15/53

📁문제 설명

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

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

새로운 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

📁입출력 예

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

📁풀이

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

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

    return answer

이번 문제는 다른 사람의 풀이를 보고 해결할 수 있었다. 심지어 이해하는데도 오래 걸림...

핵심은 n*n을 반복해서 넣기에는 사이즈가 너무 커 시간초과가 난다.
그래서 left, right+1의 사이값, 즉 인덱스(index)를 반복하며 계산했다.

몫과 나머지를 이용해 해당 위치에 있는 값을 찾아내는 것이 신기했다.(어떻게 찾은거지...)
나는 아직 갈길이 멀다. 더 열심히 해야겠다.

0개의 댓글