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

문제

배열의 길이에 따라 다른 연산하기 : 문제 링크

문제 분석

• 정수 배열 arr과 정수 n이 매개변수로 주어진다. arr의 길이가 홀수라면 arr의 모든 짝수 인덱스 위치에 n을 더한 배열을, arr의 길이가 짝수라면 arr의 모든 홀수 인덱스 위치에 n을 더한 배열을 return
• if ~ else 문을 통해 arr의 길이가 홀수 및 짝수 일 경우를 분리하고, 홀수일 경우에는 for loop의 초기식을 0으로 설정하고, 증가식을 += 2로 설정하여 짝수 인덱스의 원소를 n만큼 더하여 저장. 짝수일 경우엔 for loop의 초기식을 1로 설정하고, 마찬가지로 증가식을 += 2로 설정하여 홀수 인덱스의 원소를 n만큼 더하여 저장. 최종적으로 저장된 정수 배열 arr를 return

풀이

#include <vector>

using namespace std;

vector<int> solution(vector<int> arr, int n) {
    if(arr.size() % 2 == 1) {
        for(int i = 0; i < arr.size(); i += 2) {
            arr[i] += n;
        }
    }
    else {
        for(int i = 1; i < arr.size(); i += 2) {
            arr[i] += n;
        }
    }
    return arr;
}