프로그래머스 - 배열 만들기 1

 

stride 를 사용하면 for문을 사용하는 대신 코드를 좀 더 간결하게 만들 수 있습니다.

stride는 by를 사용하여 원하는 숫자만큼 즉 기존값 += k 으로 값을 증가시킬 수 있습니다.

import Foundation

func solution(_ n:Int, _ k:Int) -> [Int] {
        return Array(stride(from: k, to: n+1, by: k))
}

 

Array 를 사용하는 대신 map를 사용하여 배열을 만드는 방법도 있습니다.

func solution(_ n:Int, _ k:Int) -> [Int] {
    return stride(from: k, to: n+1, by: k).map { $0 }
}