[프로그래머스 LV0] 숫자 찾기

1. 문제 설명

숫자 찾기

2. 문제 분석

num의 숫자 중에 k가 있으면 그 숫자가 있는 자릿수를 반환하고 k가 없으면 -1을 반환해준다.

3. 나의 풀이

import Foundation

func solution(_ num:Int, _ k:Int) -> Int {
    var arr: [Int] = []
    
    for i in String(num) {
        if String(i) == String(k) {
            arr.append(1)
        } else {
            arr.append(-1)
        }
    }
    return arr.contains(1) ? arr.firstIndex(of: 1)! + 1 : -1
}

num의 숫자 요소와 k의 숫자를 비교해서 같으면 빈 배열에 1을 넣어주고 같지 않으면 -1을 넣어주어 배열에 1이 있으면 1이 있는 인덱스를 반환하고 없으면 -1을 반환해준다.

동일 값이 있을 수 있으므로 firstIndex를 사용하였다.

firstIndex를 사용하면 제일 앞자리에 인덱스를 반환해주고 옵셔널이 반환된다.