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

문제

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

문제 분석

• 정수 배열 arr이 주어진다. arr의 각 원소에 대해 값이 50보다 크거나 같은 짝수라면 2로 나누고, 50보다 작은 홀수라면 2를 곱한다. 그 결과인 정수 배열을 return
• for loop에서 if ~ else문을 통해 50보다 크거나 같은 짝수일 경우에는 원소에 2로 나누고, 50보다 작은 홀수일 경우엔 2를 곱하여 answer에 저장. 이외의 경우는 원소 그대로 answer에 저장. 최종적으로 저장된 answer 벡터를 return

풀이

#include <vector>

using namespace std;

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