Your Name

I am an amazing person.

1 분 소요

문제 설명

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

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

제한 사항

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

입출력 예시

arr n result
[49, 12, 100, 276, 33] 27 [76, 12, 127, 276, 60]
[444, 555, 666, 777] 100 [444, 655, 666, 877]

입출력 예 설명 1

  • 예제 1번의 arr의 길이는 5로 홀수입니다. 따라서 arr의 짝수 인덱스 0, 2, 4에 주어진 n 값인 27을 더하면 [76, 12, 127, 276, 60]이 됩니다. 따라서 [76, 12, 127, 276, 60]를 return 합니다.

입출력 예 설명 2

  • 예제 2번의 arr의 길이는 4로 짝수입니다. 따라서 arr의 홀수 인덱스 1, 3에 주어진 n 값인 100을 더하면 [444, 655, 666, 877]이 됩니다. 따라서 [444, 655, 666, 877]를 return 합니다.

코드 구현

Python

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

시간 복잡도 - Python

  • 루프의 절반을 순회하므로 시간 복잡도는 $O(n)$이다.

개선할 점 - Python

  • for 문을 하나로 만들어 코드의 길이를 줄일 수 있다.
def solution(arr, n):
    start = 1 if len(arr) % 2 == 0 else 0
    for i in range(start, len(arr), 2):
        arr[i] += n
    return arr

Java

class Solution {
    public int[] solution(int[] arr, int n) {
        if (arr.length % 2 == 0) {
            for (int i = 1; i < arr.length; i += 2) {
                arr[i] += n;
            }
        } else {
            for (int i = 0; i < arr.length; i += 2) {
                arr[i] += n;
            }
        }

        return arr;
    }
}

시간 복잡도 - Java

  • 루프의 절반을 순회하므로 시간 복잡도는 $O(n)$이다.

개선할 점 - Java

  • for 문을 하나로 만들어 코드의 길이를 줄일 수 있다.
class Solution {
    public int[] solution(int[] arr, int n) {
        int start = (arr.length % 2 == 0) ? 1 : 0;
        for (int i = start; i < arr.length; i += 2) {
            arr[i] += n;
        }
        return arr;
    }
}

참고