[프로그래머스] 배열의 길이에 따라 다른 연산하기 (Level 0, JavaScript)

❗ 문제

출처: 프로그래머스 코딩 테스트 연습,

https://school.programmers.co.kr/learn/courses/30/lessons/181854

프로그래머스

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

 

 

❓ 나의 풀이

function solution(arr, n) {
    let answer = [];
    
    for (let i = 0; i < arr.length; i++) {
        if (arr.length % 2 === 1) {
            if (i % 2 === 0) {
                answer.push(arr[i] + n);
            } else {
                answer.push(arr[i]);
            }
        } else if (arr.length % 2 === 0) {
            if (i % 2 === 1) {
                answer.push(arr[i] + n);
            } else {
                answer.push(arr[i]);
            }
        }
    }
    return answer;
}

우선, for 문을 사용하여 arr의 각 요소를 확인하고

arr의 길이가 홀수인지 짝수인지 확인하기 위해 if-else문을 사용했다.

 

만약, arr의 길이가 홀수라면 짝수 인덱스에 해당하는 요소에만 n을 더하고

홀수인덱스에 해당하는 요소는 그대로 answer에 추가된다.

arr의 길이가 짝수라면 홀수 인덱스에 해당하는 요소에만 n을 더하고

짝수 인덱스에 해당하는 요소는 그대로 answer에 추가된다.