본문 바로가기
코딩테스트 준비/프로그래머스

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

by 스테디코디스트 2023. 10. 16.
반응형

<문제 소개>


<소스 코드>

#include <string>
#include <vector>

using namespace std;

vector<int> solution(vector<int> arr, int k) 
{
    if(k % 2 == 0)
    {
        // k가 짝수
        for(int i = 0; i < arr.size(); i++)
        {
            arr[i] += k;
        }
    }
    else
    {
        // k가 홀수
        for(int i = 0; i < arr.size(); i++)
        {
            arr[i] *= k;
        }
    }
    
    return arr;
}

<풀이과정>

1. k를 2로 나눈 나머지 값을 이용해 짝수인지 홀수인지 판단한다.

2. 짝수라면 주어진 arr를 반복하며 각 원소에 k만큼 더해준다.

3. 홀수라면 주어진 arr를 반복하며 각 원소에 k만큼 곱해준다.

4. arr를 리턴해준다.


<코멘트>

어렵지 않게 금방 풀었다.


<제출결과>