divisor로 나누어 떨어지는 값을 오름차순으로 정렬한 배열을 반환하는 함수 구현하기
array의 각 element 중 divisor로 나누어 떨어지는 값을 오름차순으로 정렬한 배열을 반환하는 함수, solution을 작성해주세요.
divisor로 나누어 떨어지는 element가 하나도 없다면 배열에 -1을 담아 반환하세요.
arr | divisor | return |
[5, 9, 7, 10] | 5 | [5, 10] |
[2, 36, 1, 3] | 1 | [1, 2, 3, 36] |
[3,2,6] | 10 | [-1] |
divisor
로 나누어지는지 확인한다.
-1
을 담고, 반환한다.
function solution(arr, divisor) {
let answer = [];
arr.forEach((num) => {
if (num % divisor == 0) {
answer.push(num);
}
});
if (answer.length == 0) {
return [-1];
}
answer.sort(function (a, b) {
return a - b;
});
return answer;
}
def solution(arr, divisor):
answer = []
for num in arr:
if num % divisor == 0:
answer.append(num)
if not answer:
return [-1]
answer.sort()
return answer
더 간소화한다면, 아래와 같이 구현할 수 있다.
def solution(arr, divisor):
answer = sorted([num for num in arr if num % divisor == 0])
return answer if answer else [-1]