Lv0 조건에 맞게 수열 변환하기 3

 

프로그래머스

평소에 파이썬과 js를 사용해서 풀다가 C++을 배워야할거 같아서 이번문제는 C++을 이용해서 풀었다.

문제가 쉬운거라 간단하게 풀 수 있었다.

#include <string>
#include <vector>

using namespace std;

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

다른 풀이를 봤는데, 다른건 예상한 답변인데 하나가 처음보는 함수를 사용해서 풀었다.

#include <algorithm>
#include <vector>
using namespace std;

vector<int> solution(vector<int> v, int k) {
    transform(v.begin(),v.end(),v.begin(),[k](int x){return k&1?x*k:x+k;});
    return v;
}

Algorithm헤더 파일에 있는 transform을 사용한 것이다. 처음 보기도 했고, 신기하기도 해서 공부를 했는데 간단했다.

template <class InputIt1, class InputIt2, class OutputIt, class BinaryOperation>
// InputIt1 시작
// InputIt2 끝
// OutputIt 원하는 시작 위치
// BinaryOperation 적용하고 싶은 함수

이렇게 이해할 수 있었다.