[Programmers - lv01] 나누어 떨어지는 숫자 배열 (cpp / python)

나누어 떨어지는 숫자 배열

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

문제 설명

.

문제 풀이

소스코드 : C++

#include <iostream>
#include <string>
#include <vector>
#include <algorithm>

using namespace std;

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

소스코드 : Python

return에 or 처럼 조건을 줄 수 잇는지 처음 알았다.

def solution(arr, divisor):
    return sorted([ans for ans in arr if ans%divisor == 0]) or [-1] # 앞의 조건이 거짓일 때 뒤에 것 반환