[프로그래머스 LV.0] 숫자 찾기

프로그래머스 LV.0 모음

숫자 찾기

문제 설명

정수 num과 k가 매개변수로 주어질 때, num을 이루는 숫자 중에 k가 있으면 num의 그 숫자가 있는 자리 수를 return 하고 없으면 -1을 return 하도록 solution 함수를 완성해 보세요.

제한 사항

• 0 < num < 1,000,000
• 0 ≤ k < 10
• num에 k가 여러 개 있으면 가장 처음 나타나는 자리를 return 합니다.

입출력 예

num	k	reulst
29183	1	3
232443	4	4
123456	7	-1

입출력 예 설명

입출력 예 #1

• 29183에서 1은 3번째에 있습니다.

입출력 예 #2

• 232443에서 4는 4번째에 처음 등장합니다.

입출력 예 #3

• 123456에 7은 없으므로 -1을 return 합니다.

---

제출

import Foundation

func solution(_ num:Int, _ k:Int) -> Int {
    let n = String(num).map{Int(String($0))!}
    return n.contains(k) ? n.firstIndex(of: k)! + 1 : -1
}

입력받은 정수를 Int형 배열로 변환한 후, k 가 존재하면 인덱스를, 없다면 -1을 반환한다.