Functional Swift: List Replication : read an array and number of times and return a new array with each element that number of times.

https://www.hackerrank.com/challenges/fp-list-replication/problem


import Foundation

func printElements(_ numberOfTimes: Int, _ list: [Int]) -> [Int] {
    if list.count <= 0 {
        return list
    }

    let appendedArray = addToList(numberOfTimes, list.first!)
    let newArray = Array(list.dropFirst())
    return appendedArray + printElements(numberOfTimes, newArray)
}

func addToList(_ numberOfTimes: Int, _ number: Int)  -> [Int] {
   if numberOfTimes <= 0 {
       return []
   } 

   let list = [number] 
   return list + addToList(numberOfTimes - 1, number)
}

print("\(printElements(5, [3, 1, 2, 4]))")



---------------------------------------------------------------------------------------
                REFACTORED / UPDATED to reduce variable names 
---------------------------------------------------------------------------------------

import Foundation


func printElements(_ numberOfTimes: Int, _ list: [Int]) -> [Int] {
    if list.count <= 0 {
        return list
    }

    let appendedArray = addToList(numberOfTimes, list.first!)
    let newArray = Array(list.dropFirst())
    return appendedArray + printElements(numberOfTimes, newArray)
}

func addToList(_ numberOfTimes: Int, _ number: Int)  -> [Int] {
   if numberOfTimes <= 0 {
       return []
   } 

   return [number] + addToList(numberOfTimes - 1, number)
}

print("\(printElements(5, [3, 1, 2, 4]))")