본문 바로가기
Programming Practice/Python

프로그래머스 <배열의 길이에 따라 다른 연산하기>

by TAMIK 2025. 4. 16.
728x90
def solution(arr, n):
    if len(arr) % 2 == 0:
        for i in range(1, len(arr) + 1, 2):
            arr[i] += n
        return arr
    else:
        for i in range(0, len(arr) + 1, 2):
            arr[i] += n
        return arr

 

 

좀 더 간결하게

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