[Swift] 프로그래머스 입문 문제, k의 개수

문제

 

풀이 방법

1. i ~ j 까지 Int 배열을 만든다
2. 고차함수 map을 이용해 String 배열로 만듬
3. flatmap을 이용하면 ["11"]을 "1", "1"로 리턴해주기 때문에 flatmap 을사용
4. filter함수를 이용하고
5. 마무리 count

 

코드

import Foundation

func solution(_ i:Int, _ j:Int, _ k:Int) -> Int {
    
    return Array(i...j).map {
        String($0)
    }.flatMap {
        String($0).map {
            String($0)
        }
    }.filter {
        $0 == String(k)
    }.count
}