euisblue
12910. 나누어 떨어지는 숫자 배열

Level 1 나누어 떨어지는 숫자 배열

시간 복잡도: O(Nlogn)

1#include <bits/stdc++.h> 2 3using namespace std; 4 5vector<int> solution(vector<int> arr, int divisor) { 6 vector<int> answer; 7 const int SIZE = arr.size(); 8 9 for(int i=0; i<SIZE; ++i) { 10 if(arr[i]%divisor==0) answer.push_back(arr[i]); 11 } 12 13 sort(answer.begin(), answer.end()); 14 15 if(answer.size()==0) answer.push_back(-1); 16 return answer; 17}

배열의 각 원소들이 주어진 divisor와 나누어 떨어지는지 확인한다. 마지막으로 정답 배열을 오름순으로 정렬해주어야 하기 때문에 std::sort()를 사용했다.