coding study/programmars

[level 0] n의 배수 고르기

yeoonii 2024. 5. 27. 20:05

문제 설명

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


제한사항
  • 1 ≤ n ≤ 10,000
  • 1 ≤ numlist의 크기 ≤ 100
  • 1 ≤ numlist의 원소 ≤ 100,000

def solution(n, numlist):
    ans = []
    for i in range(len(numlist)):
       if numlist[i] % n == 0:
        ans.append(numlist[i])
    return ans

 

다른 사람의 풀이

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]