[Level.0] n의 배수 고르기

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

제한사항

• 1 ≤ n ≤ 10,000
• 1 ≤ numlist의 크기 ≤ 100
• 1 ≤ numlist의 원소 ≤ 100,000

문제 해결

def solution(n, numlist):
    answer = []
    for num in numlist:
        if num%n == 0:
            answer.append(num)
    return answer

def solution(n, numlist):
    answer = [num for num in numlist if num%n==0]
    return answer

 

주석: n의 배열이 아닌 값을 직접 numlist에서 삭제시 numlist의 인덱스 값과 배열이 흐트러져 값이 제대로 삭제가 안됨.. 그러므로 차라리 새로운 배열에 그 원하는 값을 삽입하는게 더 좋다.