[프로그래머스] N의 배수 고르기

allnight5·2022년 11월 16일
0

프로그래머스 입문

목록 보기
21/53

정수 n과 정수 배열 numlist가 매개변수로 주어질 때, numlist에서 n의 배수가 아닌 수들을 제거한 배열을 return하도록 solution 함수를 완성해주세요.

자바

---------------------------------

import java.util.Arrays;
class Solution {
    public int[] solution(int n, int[] numlist) {
        int[] answer = {};
        answer = Arrays.stream(numlist).filter(i ->i%n ==0).toArray();
        System.out.print(Arrays.toString(answer));
        return answer;
    }
}

파이썬

---------------------------------

def solution(n, numlist):

answer = [i for i in numlist if i%n==0]
return answer

---------------------------------

def solution(n, numlist):

return list(filter(lambda v: v%n==0, numlist))

---------------------------------

def solution(n, numlist):

return [num for num in numlist if num % n == 0]
profile
공부기록하기

0개의 댓글