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

 

출처: 프로그래머스 코딩 테스트 연습, https://programmers.co.kr/learn/challenges

 

 

 

월간코드챌린지 때 효율성에서 막혀서 결국 못 풀었는데 풀이 보고 너무나도 간단해서 놀랐다.

 

1차원 배열을 n 으로 나눈 몫, 나머지에서 max 값만 구하면 되다니..

 

** python

 

 

 

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

 

 

 

참고: https://sangsangss.tistory.com/197

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