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

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

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

문제 접근법

• divisor값이 1일 경우 arr의 배열 값들이 나누어 떨어지므로 arr을 sort하여 리턴해줍니다.
• divisor값이 1이 아닌 경우 arr배열을 순회하면서 arr배열 원소 값들을 divisor 값과 %연산을 통하여 0이 나오는지 확인합니다.
• 0이 나올경우 나누어 떨어지는 값이므로, 해당 값을 저장해줍니다.
• 순회가 끝난 후 저장한 값들을 sort를 통해 오름차순으로 정렬 후 리턴해줍니다.

아래는 코드입니다.

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