Algorithm/프로그래머스
[프로그래머스] n^2 배열 자르기 - 파이썬
potato_pizza
2024. 6. 19. 00:38
728x90
n^2 배열 자르기
https://school.programmers.co.kr/learn/courses/30/lessons/87390
문제
입출력 예 설명
입출력 예 #1
입출력 예 #2
코드
- 시간 초과
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)
return answer[left:right+1]
- 몫, 나머지를 활용하여 각 인덱스의 숫자를 찾는 방법
def solution(n, left, right):
answer = []
for i in range(left, right+1):
a = i // n
b = i % n
if a < b:
a, b = b, a
answer.append(a+1)
return answer
728x90
반응형