본문 바로가기
algorithm

[프로그래머스]숫자 찾기 - java, python

by skysline 2023. 3. 9.
반응형

`숫자 찾기`

문제 설명

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

입출력 예


입출력 예 #1

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

입출력 예 #2

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

입출력 예 #3

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

 

java code

import java.util.ArrayList;
import java.util.List;
class Solution {
    public int solution(int num, int k) {
        int answer = 0;
        String snum = Integer.toString(num);

        List<Character> lst = new ArrayList<>();
        for (char item: snum.toCharArray()) {
            lst.add(item);
        }
        answer = lst.indexOf(Character.forDigit(k,10));
        if (answer == -1)
            answer += -1;
        return answer+1;
    }
}
  • 다른풀이
public int solution(int num, int k) {
        int answer = 0;
        List<String[]> lst = new ArrayList<>();
        String Ks = String.valueOf(num);
        String Sk = String.valueOf(k);
        answer = Ks.indexOf(Sk);
        
        return answer<0 ? -1:answer+1;
    }

python code

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

반응형

댓글