배열 만들기1
Swift
import Foundation
func solution(_ n: Int, _ k: Int) -> [Int] {
let numbers = Array(1...n)
let result = numbers.filter { k * $0 < n + 1 }.map { k * $0 }
return result
}
Swift
복사
filter 함수를 사용하여 k * i < n + 1 조건을 충족하는 요소만 남기고,
map 함수를 사용하여 해당 요소 각각에 k를 곱한다.
Python
def solution(n, k):
return [k * i for i in range(1, n+1) if k * i < n+1]
Python
복사