[프로그래머스] 나누어 떨어지는 숫자 배열 C++ 풀이

https://programmers.co.kr/learn/courses/30/lessons/12910?language=cpp 

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

 

전체 코드

#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);
    else
        sort(answer.begin(),answer.end());
    
    return answer;
}

 

divisor로 나누어 떨어지는 수는 answer에 넣은 뒤

정렬하여 출력을 하면 됩니다.

하지만, divisor로 나누어 떨어지는 수가 없는 경우도 있기에 answer 크기가 0이라면 -1을 넣어 줍니다.