문제
접근
arr를 순회하며 나머지가 0인 수들만 따로 저장해두고 이를 정렬하면 된다.
코드
- C++ 코드
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
#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;
}
- 파이썬 코드
1
2
3
4
5
6
7
8
9
10
def solution(arr, divisor):
answer = []
for item in arr:
if item % divisor == 0:
answer.append(item)
if len(answer) == 0:
answer.append(-1)
else:
answer.sort()
return answer