[프로그래머스 level2 n^2 배열 자르기] list comprehension 사용해보기

해결해야하는 문제는 다음과 같았다.

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

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

 

처음으로 했던 아이디어는, 문제를 보고 차근차근 그냥 따라하였다.

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

--> n 행 n열 을 조건에 따라 list comprehension을 이용하여 만들었다.

arr2d = [[max(i,j) for i in range(1,n+1)] for j in range(1,n+1)]

 

 

2. 1행, 2행, ..., n행을 잘라내어 모두 이어붙인 새로운 1차원 배열을 만듭니다.

--> 이 또한 list comprehension을 이용하여 2차원 배열을 1차원으로 만들었다.

answer = [data for inner_list in arr2d for data in inner_list]

 

3. 새로운 1차원 배열을 arr이라 할 때, arr[left], arr[left+1], ..., arr[right]만 남기고 나머지는 지웁니다.

--> list slicing을 통해 주어진 범위만 남겨서 return 했다.

return answer[left:right+1]

 

 

 

다음은 시간 초과가 났던 코드이다.

def solution(n, left, right):
    arr2d = [[max(i,j) for i in range(1,n+1)] for j in range(1,n+1)]
    answer = [data for inner_list in arr2d for data in inner_list]
    return answer[left:right+1]

 

 

 

범위가 1 ≤ n ≤ 10^7 이기 때문에, 시간 초과가 나는 것이기 때문에, 답으로 가능한 행에 대해서만 2차원 배열을 만든 뒤,  left~right 에 대한 배열을 찾으면 다음과 같다.

def solution(n, left, right):
    left_row = left// n + 1
    left_col = left % n
    right_row = right // n + 1
    right_col = right % n
    arr2d = [[max(i,j) for i in range(1,n+1)] for j in range(left_row,right_row+1)]

    answer = [data for inner_list in arr2d for data in inner_list]
    return answer[left_col:(right_row-left_row)*n+right_col+1]