[Swift] 프로그래머스 입문 문제, 숫자 찾기

문제

 

풀이 방법

1. 첫번째 함수 인자인 num을 compactMap을 이용해 배열로 변경
2. firstIndex를 통해 k가 첫번째로 포함된 인덱스를 찾음
3. 못 찾는경우 else문으로 빠지기 때문에 -1을 return
4. 인덱스를 찾을경우 +1을 더해서 return

 

코드

import Foundation

func solution(_ num:Int, _ k:Int) -> Int {
    
    let digits = String(num).compactMap {
        $0.wholeNumberValue
    }
    
    guard let firstIndex = digits.firstIndex(of: k) else {
        return -1
    }
    
    return firstIndex + 1
}