[swift] 프로그래머스 - 잘라서 배열로 저장하기

프로그래머스

---

문제풀이를 직관적으로 이해할 수 있도록 고차함수를 사용하지 않은 풀이와
고차함수를 사용한 풀이로 나누어 공부하고 있습니다.

 

고차함수를 사용하지 않은 풀이

func solution(_ my_str:String, _ n:Int) -> [String] {
    var lower = 0
    var upper = n - 1
    var result = [String]()
    for _ in 1...my_str.count/n {
        result.append(String(my_str[my_str.index(my_str.startIndex, offsetBy: lower)...my_str.index(my_str.startIndex, offsetBy: upper)]))
        lower += n
        upper += n
    }
    if my_str.count % n != 0 {
        result.append(String(my_str.suffix(my_str.count % n)))
    }
    return result
}

 

다른 방식의 풀이

func solution(_ my_str:String, _ n:Int) -> [String] {
    var result: [String] = []
    var split = ""
    for (index, item) in my_str.enumerated() {
        split += String(item)
        if split.count == n {
            result.append(split)
            split = ""
        }
    }
    if !split.isEmpty {
        result.append(split)
    }
    return result
}

 

고차함수를 사용한 풀이

func solution(_ my_str:String, _ n:Int) -> [String] {
    
    stride(from: 0, to: my_str.count, by: n).map {
        let start = my_str.index(my_str.startIndex, offsetBy: $0)
        let end = my_str.index(start, offsetBy: n, limitedBy: my_str.endIndex) ?? my_str.endIndex
        return String(my_str[start..<end])
    }
}