[프로그래머스 / C++] 조건에 맞게 수열 변환하기 3

문제

조건에 맞게 수열 변환하기 3 : 문제 링크

문제 분석

• 정수 배열 arr와 자연수 k가 주어진다. 만약 k가 홀수라면 arr의 모든 원소에 k를 곱하고, k가 짝수라면 arr의 모든 원소에 k를 더한다. 이러한 변환을 마친 후에 arr를 return
• for loop에서 if ~ else문을 통해 k가 홀수일 경우, 해당 원소에 k값을 곱한값을, 짝수일 경우, 해당 원소에 k값을 더한값을 answer에 저장. 최종적으로 저장된 answer를 return

풀이

#include <vector>

using namespace std;

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