나누어 떨어지는 숫자 배열

들어가는 말

정수 divisor을 제공할 때, divisor로 나눠 떨어지는 값들을 오름차순으로 정렬하는 문제이다. 만일 나눠 떨어지는 값이 없다면 배열에다가 -1을 넣어준다.

코드 설명

일단 주어진 배열을 전부 돌면서 divisor로 나눠 떨어지는 값을 answer 배열에다가 넣어 주었다.
그 다음 나눠 떨어지는 값이 모두 없을 경우에는 -1을 삽입한다.
마지막으로 정답 배열을 정렬해서 출력하면 끝이다. 정렬할 때는 드디어 sort()함수를 썼다!! 컴파일 에러 없이 완벽하게!!! 신난다!!!

코드

#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;
}