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

Problem

Solution

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

문제 조건을 확인하면 2차 배열의 i행의 원소들은 i열까지는 i의 값을 가지고 그 이후로 1씩 더해져 나간다.

따라서 주어진 left, right 리스트에서 몇 행, 몇 열인지 구한 뒤 계산하여 리스트에 더해주었다.

Reference

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