[Level.0] 배열의 길이에 따라 다른 연산하기

문제 설명: 정수 배열 arr과 정수 n이 매개변수로 주어집니다. arr의 길이가 홀수라면 arr의 모든 짝수 인덱스 위치에 n을 더한 배열을, arr의 길이가 짝수라면 arr의 모든 홀수 인덱스 위치에 n을 더한 배열을 return 하는 solution 함수를 작성해 주세요.

제한사항

• 1 ≤ arr의 길이 ≤ 1,000
• 1 ≤ arr의 원소 ≤ 1,000
• 1 ≤ n ≤ 1,000

문제 해결

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

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

 

주석: 이 로직의 시간 복잡도는 정확히 O(N), if-else 분기 처리에 의해 두 개의 for문 중 단 하나만 실행되기 때문임, 또한, 각 루프는 전체 배열 길이 N의 절반인 N/2번만큼만 반복하므로 엄밀히 말하면 연산 횟수는 N/2에 가깝다. 그러나 빅오 표기법에서는 상수항을 무시하므로 최종적으로는 O(N)으로 표기하는 것이 맞다.