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

문제

프로그래머스

접근 방식

0) 위치값

#0	#1	#2	#3
#4	#5	#6	#7
#8	#9	#10	#11
#12	#13	#14	#15

 

1) 열(column) ➡️ 위치값 % n + 1

1	2	3	4
1	2	3	4
1	2	3	4
1	2	3	4

 

2) 행(row) ➡️ 위치값 // n + 1

1	1	1	1
2	2	2	2
3	3	3	3
4	4	4	4

 

3) 열과 행 중 더 큰 값 ➡️ max( 위치값 % n ,  위치값 // n ) + 1

1	2	3	4
2	2	3	4
3	3	3	4
4	4	4	4

 

🔶a행 b열의 값은 max(a,b)이다.

풀이

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

실행결과