웅재의 코딩세상

[프로그래머스] 나누어 떨어지는 숫자 배열 본문

코딩테스트/프로그래머스 - LV 1

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

웅드 2024. 2. 9. 17:26

 

  • 풀이
#include <string>
#include <vector>
#include <algorithm>

using namespace std;

vector<int> solution(vector<int> arr, int divisor) {
    vector<int> answer;
    int v=0;
    for(int i=0; i<arr.size();i++){
        if(arr[i] % divisor == 0) {
            answer.push_back(arr[i]);
            v++;
        }
    }
    if(v==0) answer.push_back(-1);
    sort(answer.begin(),answer.end());
    return answer;
}
// return answer.empty() ? vector<int>(1, -1) : answer; empty함수 활용 가능
반응형