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

문제 링크 https://programmers.co.kr/learn/courses/30/lessons/12903

문제

array의 각 element 중 divisor로 나누어 떨어지는 값을 오름차순으로 정렬한 배열을 반환하는 함수, solution을 작성해주세요. divisor로 나누어 떨어지는 element가 하나도 없다면 배열에 -1을 담아 반환하세요.

제한 조건

• arr은 자연수를 담은 배열입니다.
• 정수 i, j에 대해 i ≠ j 이면 arr[i] ≠ arr[j] 입니다.
• divisor는 자연수입니다.
• array는 길이 1 이상인 배열입니다.

풀이 과정

나누어 떨어진다면 answer 벡터에 넣고, 오름차순으로 정렬해주면 된다.

풀이1

#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.size() == 0) answer.push_back(-1);
    else sort(answer.begin(), answer.end());
    return answer;
}

풀이2

#include <string>
#include <vector>

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.size() == 0) answer.push_back(-1);
    else {
        // sort함수 오름차순 정렬
        for(int i = 0; i < answer.size(); i++){
            for(int j = 0; j < answer.size(); j++){
                if(answer[i] < answer[j]){
                    int temp = answer[i];
                    answer[i] = answer[j];
                    answer[j] = temp;
                }
            }
        }
    }
    return answer;
}