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

문제

---

문제를 보시려면 링크를 클릭해주세요.

풀이

---

1. arr 요소가 divisor로 나누어 떨어질 때, 해당 값을 answer 벡터에 담아줍니다.

2. answer 벡터를 오름차순 정렬합니다.

3. answer 벡터가 비어있을땐 나누어 떨어지는 요소가 없는 경우이므로 -1을 담아줍니다.

소스 코드

---

#include <algorithm>
#include <vector>

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