오답노트

프로그래머스 1단계 - 나누어 떨어지는 숫자 배열 본문

C,C++/코딩테스트

프로그래머스 1단계 - 나누어 떨어지는 숫자 배열

권멋져 2022. 5. 27. 18:00
https://programmers.co.kr/learn/courses/30/lessons/12910?language=cpp 
 

코딩테스트 연습 - 나누어 떨어지는 숫자 배열

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

programmers.co.kr

- 문제파악

정수 배열과 나눗셈을 하기위한 정수가 주어지는데 배열의 요소들 중에서 나눗셈을 하기 위한 정수로 나누어지는 수를 오름차순으로 정렬한 배열을 출력하라, 만약 나누어지는 수가 없다면 -1을 요소로 가지는 배열을 출력하라

 

- 정답

#include "bits/stdc++.h"

using namespace std;

vector<int> solution(vector<int> arr, int divisor) {
    vector<int> answer;
    
    for(auto a : arr)
    {
        if(a%divisor == 0)
            answer.push_back(a);
    }
    
    if(answer.size())
        sort(answer.begin(),answer.end());
    else
        answer.push_back(-1);
    
    
    return answer;
}