Lv2 프로그래머스(Programmers)[Python][파이썬] n^2 배열 자르기

"""
출처:프로그래머스,
https://school.programmers.co.kr/learn/courses/30/lessons/87390

"""
# 풀이 과정
count = left
result = []
k = 1 # 내부에 들어갈 수
num = left % n + 1 # 내부 체크 수

while count <= right:
if int(count / n) + 1 >= num and num <= n:
k = int(count / n) + 1
if count >= left and count <= right:
result.append(k)
num += 1
count += 1

elif int(count / n) + 1 < num and num <= n:
if count >= left and count <= right:
result.append(num)
num += 1
count += 1

if num > n:
num = 1

return result