[프로그래머스] Lv2 - n^2 배열 자르기 (파이썬)

_temp 2022. 5. 13. 10:48

코딩테스트 연습 - n^2 배열 자르기

정수 n, left, right가 주어집니다. 다음 과정을 거쳐서 1차원 배열을 만들고자 합니다. n행 n열 크기의 비어있는 2차원 배열을 만듭니다. i = 1, 2, 3, ..., n에 대해서, 다음 과정을 반복합니다. 1행 1열부

programmers.co.kr

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

풀이
1. left부터 right까지 해당 index가 2차원 배열이었을 때의 (인덱스 값들 중 최댓값+1)을 arr에 append
2. arr 반환

구하는 것은 left~right의 1차원 배열이다.

따라서 left~right을 i로 놓고 해당 배열이 2차원이었을 때의 인덱스를 구한다. (i//n, i%n)

해당 인덱스의 값을 이용해서 결괏값을 알 수 있고 그를 arr에 순서대로 추가

2차원 배열과 1차원 배열의 인덱스 값은 (i//n, i% n)과 같은 관계가 있음을 알고 있어야 한다.