C++/programmers 코딩테스트(level 1) C++

[C/C++] programmers Level 1 나누어 떨어지는 숫자 배열

CE : 하랑 2023. 9. 18. 11:26

 

 

 

 

 

코드

 

#include <vector>
#include <algorithm> // sort

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