코딩테스트/프로그래머스

n의 배수 고르기 [Swift, python]

clamp 2023. 1. 13. 15:04

문제 설명

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


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

입출력 예nnumlistresult
3 [4, 5, 6, 7, 8, 9, 10, 11, 12] [6, 9, 12]
5 [1, 9, 3, 10, 13, 5] [10, 5]
12 [2, 100, 120, 600, 12, 12] [120, 600, 12, 12]

 

def solution(n, numlist):
    answer = []

    for i in numlist:
        if i % n == 0:
            answer.append(i)
    return answer
import Foundation

func solution(_ n:Int, _ numlist:[Int]) -> [Int] {
    var result = [Int]()
    for i in numlist{
        if i % n == 0{
            result.append(i)
        }
    }
    return result
}