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

관련 문제

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

프로그래머스

 

코드

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

 

해결 포인트

기존 코드 : 시간 초과

def solution(n, left, right):
    answer = []
    for i in range(n):
        for j in range(n):
            if i >= j:
                answer.append(i+1)
            else:
                answer.append(j+1)
            if len(answer) > right+1:
                break    
        if len(answer) > right+1:
            break
    return answer[left:right+1]

핵심 문제 : 시간 초과

right보다 전체 배열이 크면 종료했지만 그럼에도 시간이 초과됨. 따라서 left ~ right까지민 반복문을 사용해야한다.