[프로그래머스] 숫자 찾기 파이썬

xianxbabx·2023년 1월 22일
1

알고리즘

목록 보기
7/14

문제 설명

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

제한사항

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

입출력 예

numkresult
2918313
23244344
1234567-1

입출력 예 설명

입출력 예 #1

  • 29183에서 1은 3번째에 있습니다.
    입출력 예 #2

  • 232443에서 4는 4번째에 처음 등장합니다.
    입출력 예 #3

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

나의 코드

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

풀이

find 함수를 이용해 원하는 위치를 구해낸다. 숫자이지만 문자로 취급하고 풀어야 하기 때문에 str() 해주는 것은 필수이다.

0개의 댓글