[프로그래머스 C++]/LEVEL 0

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

AKI(JUNI) 2024. 1. 16. 00:26

◈ 문제 설명


◈ 문제 설명 링크
코딩테스트 연습 - 배열의 길이에 따라 다른 연산하기 | 프로그래머스 스쿨 (programmers.co.kr)

 

프로그래머스

코드 중심의 개발자 채용. 스택 기반의 포지션 매칭. 프로그래머스의 개발자 맞춤형 프로필을 등록하고, 나와 기술 궁합이 잘 맞는 기업들을 매칭 받으세요.

programmers.co.kr


◈ 작성 코드

#include <string>
#include <vector>

using namespace std;

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