[프로그래머스 / Java] [level 0] 배열의 길이에 따라 다른 연산하기

문제 설명

정수 배열 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]

---

제출 답안 :

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

 

처음에 단순히 if문으로 배열길이가 홀수/짝수일때 index가 짝수/홀수일때로 구분지어 작성했다.

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

 

생각을 해보면 짝수/홀수의 index는 2씩 증가하기 때문에, 처음 시작하는 index가 짝수/홀수인지에 따라 패턴이 결정된다.
따라서 처음 index만 정해주면 코드를 더 간단하게 작성할 수 있다는것을 깨닳았다.