프로그래머스 알고리즘/0 & 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
반응형