[프로그래머스 스쿨] 코딩테스트 연습 기초 - 배열의 길이에 따라 다른 연산하기

문제 설명

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

입출력 예 설명

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

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

기본 코드

class Solution {
    public int[] solution(int[] arr, int n) {
        int[] answer = {};
        return answer;
    }
}

내가 구현한 코드

import java.util.Arrays;
class Solution {
    public int[] solution(int[] arr, int n) {
        if (arr.length % 2 != 0){
            for(int i = 0; i <= arr.length; i++) {
                if(i % 2 == 0) {
                    arr[i] += n;
                  }       
                }
        } else {
            for(int i = 0; i <= arr.length; i++) {
                if(i % 2 != 0) {
                    arr[i] += n;
                }
            }
         }
        int[] answer = Arrays.copyOf(arr, arr.length);
        return answer;
    } 
}

문제 해결

처음에는 변수의 길이 if문을 통해 짝수와 홀수를 구별합니다. 홀수하면 for문을 통해 배열의 원소를 가져오면서 홀수라면 짝수 원소에 n값을 더하는 것이기에 다시 if문을 활용하여 더해주었다. 짝수도 마찬가지로 배열의 원소를 for문으로 가져오고 다시 if문을 활용하여 홀수 원소에 n값을 더해주었다.

알게된 점

1. if문과 for문의 구분 확실하게 하기.