프로그래머스 - 나누어 떨어지는 숫자 배열, C++

https://programmers.co.kr/learn/courses/30/lessons/12910

코딩테스트 연습 - 나누어 떨어지는 숫자 배열

 

#include <string>
#include <vector>
#include <algorithm>

using namespace std;

vector<int> solution(vector<int> arr, int divisor) {
    vector<int> answer;
    
    for(int i=0; i<arr.size(); i++){
        if( arr[i] % divisor == 0 )
            answer.push_back( arr[i] );
    }
    if( answer.size() == 0 )
        answer.push_back(-1);
    
    sort(answer.begin(), answer.end());
    
    return answer;
}

 

arr의 요소들이 divisor과 나누어 떨어지면 answer의 배열에 그 요소를 넣어주었고, 

반복문을 빠져나와 answer 배열의 크기가 0일 경우에는 -1을 넣어서 나누어 떨어지는 요소가 없다는 것을 표시, 

 

마지막에는 sort 함수를 이용해 answer를 오름차순으로 정렬시켜주었다.