N

(프로그래머스 c++)나누어 떨어지는 숫자 배열 본문

프로그래머스 알고리즘/1단계

(프로그래머스 c++)나누어 떨어지는 숫자 배열

naeunchan 2020. 4. 23. 13:26
728x90
반응형

sort와 나머지 연산만 할 수 있으면 간단하게 풀 수 있는 문제..!

나누어 떨어지는 숫자가 1도 없으면(=answer 벡터가 비어있으면) -1을 answer에 넣어주고,

그렇지 않으면 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.empty())
        answer.push_back(-1);
    else
        sort(answer.begin(), answer.end());
    return answer;
}
728x90
반응형