[swift] 프로그래머스 - 암호해독

프로그래머스

---

---

나의 풀이

import Foundation

func solution(_ cipher:String, _ code:Int) -> String {
    var count = 1
    var result = ""
    for i in cipher {
        if count % code == 0 {
            result += String(i)
        }
        count += 1
    }
    return result
}

 

개선된 풀이

- 코드길이 감소

- 고차함수 사용

- enumerated()는 문자열을 (index, element)인 열거형으로 만들어준다.

고차함수를 사용할 때 index값은 $0.offset 또는 $0.0로 사용할 수 있고

element값은 $0.element 또는 $0.1로 사용할 수 있다.

import Foundation

func solution(_ cipher:String, _ code:Int) -> String {
    return cipher.enumerated().filter{ ($0.offset + 1) % code == 0 }.reduce("") { $0 + String($1.element) }
}