[프로그래머스] Lv.0 - 숫자 찾기.py

문제

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

제한사항

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

입출력 예시

• 입출력 예 1
  • 29183에서 1은 3번째에 있습니다.
• 입출력 예 2
  • 232443에서 4는 4번째에 처음 등장합니다.
• 입출력 예 3
  • 123456에 7은 없으므로 -1을 return 합니다.

코드

def solution(num, k):
    try:
        answer = str(num).index(str(k)) + 1
    except:
        answer = -1
    return answer

출처: 프로그래머스 코딩 테스트 연습, https://school.programmers.co.kr/learn/challenges