def solution(A, K):
num = K % len(A)
if num == 0:
return A
return A[num-1:]+A[0:num-1]
끝나면 테스트 코드를 알려주는데 example 빼고 다틀림^^
이래서 테스트 코드를 짜는게 중요하다고 하는 갑다
여러가지 에러가 있지만
ZeroDivisionError: integer division or modulo by zero
가 있으니, 예외(K = 0 or 길이 0)에 대한 처리를 해줘야겠다
def solution(A, K):
if len(A) == 0 or K == 0:
return A
if K > len(A):
K = K % len(A)
return A[-K:] + A[0:len(A)-K]
결과는 여기에
보니까 0 <= K <= 5일 때, element가 하나만 있는 케이스에서 오류가 발생한다
len(A) ==1 인 것에 대해 예외처리를 해주었다
def solution(A, K):
if len(A) == 0 or K == 0 or len(A) == 1:
return A
if K > len(A):
K = K % len(A)
return A[-K:] + A[0:len(A)-K]
아니 예외 처리 좀 잘 하자 제발
이런 문제에 시간 너무 많이 쏟았음 😑