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

❔문제설명

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

🤔아이디어

배열의 길이에 따라서 경우의 수 나누고 반복문 통해서 넣어주면되는데... 바꾸지 않아도 되는 값에 대해서는 어떻게 해야하지..?

❌틀린코드

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

🙄오류

index범위 오류가 났네... 그냥 i의 값에 대해서도 홀 짝의 경우로 나눠서 그냥 넣는 방식으로 바꿔볼까...

💡코드풀이

class Solution {
    public int[] solution(int[] arr, int n) {
        int arrLen = arr.length;
        int[] answer = new int[arrLen];//⭐크기 초기화!!
        
        if(arrLen % 2 == 0){
            for(int i = 0 ; i < arrLen ; i++){
                if(i % 2 == 0){
                    answer[i] = arr[i];
                } else {
                    answer[i] = arr[i] + n;
                }
            }
        } else {
            for(int i = 0 ; i < arrLen ; i++){
                if(i % 2 == 0){
                    answer[i] = arr[i] + n;
                } else {
                    answer[i] = arr[i];
                }
            }
        }
        return answer;
    }
}