[Programmers] 숫자 찾기

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

source: https://school.programmers.co.kr/learn/courses/30/lessons/120904

 

1.

def solution(num, k):
    for i, n in enumerate(str(num)):
        if str(k) == n:
            return i + 1
    return -1

 

 

2. str로 변수 선언하고 return 하는게 가독성 측면에서 나을 듯

def solution(num, k):
    return -1 if str(k) not in str(num) else str(num).find(str(k)) + 1