프로그래머스[Lv.0] 배열의 길이에 따라 다른 연산하기

문제설명

정수 배열 arr과 정수 n이 매개변수로 주어짐.

arr의 길이가 홀수라면 arr의 모든 짝수 인덱스 위치에 n을 더한 배열을, arr의 길이가 짝수라면 arr의 모든 홀수 인덱스 위에 n을 더한 배열을 리턴

 

코드

def solution(arr, n):
    N=len(arr)
    if N%2 == 1:
        for i in range(0,N,2):
            arr[i]+=n
    else:
        for i in range(1,N,2):
            arr[i]+=n
    return arr