[ 프로그래머스 ] 나누어 떨어지는 숫자 배열 (연습문제)
코드
#include <string>
#include <vector>
#include <algorithm> //sort를 사용하기 위해 추가
using namespace std;
vector<int> solution(vector<int> arr, int divisor) {
vector<int> answer;
int size = arr.size();
for(int i=0;i<size;i++){
if(arr[i]%divisor==0){ //나누어 떨어지는 값일 경우
answer.push_back(arr[i]);
}else if (arr[i]%divisor!=0){ //나누어 떨어지는 값이 아닐 경우
continue;
}
}
if(answer.size()==0){ //나누어 떨어지는 값이 하나도 없는 경우
answer.push_back(-1);
}
sort(answer.begin(),answer.end()); //오름차순으로 정렬
return answer;
}