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

Link

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

 

Solution

처음에 풀때는 간단한 문제라 생각하고 풀었지만 역시나 시간초과에 걸리고 말았다.

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

    for i in range(n):
        for j in range(n):
            answer.append(max(i + 1, j + 1))

    return answer[left:right + 1]

 

Other Solution

이에 left와 right의 값으로 해당 위치의 값을 찾을 방법을 고민해보기 시작했고 그 결과 위치값에서 n을 나눠준 나머지와 몫 중 큰 값이 1을 더해준 값이 해당 위치의 값이라는 사실을 알게 되었다.

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

    for i in range(left, right + 1):
        answer.append(max(i // n, i % n) + 1)

    return answer

🔑 key point: 해당 위치 값 = max(해당 위치 // n, 해당 위치 % n) + 1
📌 Tip: max(i // n, i % n)