[Python, JavaScript, Java] 숫자 찾기

프로그래머스 문제 풀이입니다. 문제는 프로그래머스에서 확인하세요!

 

Python

파이썬에는 find()라는 함수가 있어서 num과 k를 문자열로 만들어준다면 아주 쉽게 정답을 도출할 수 있다. 문제는 문자열로 찾아야하기 때문에 인덱스 번호가 하나 작게 나온다는 점.

def solution(num, k):
    strNum = str(num)
    strK = str(k)

    if(strNum.find(strK) != -1) :
        return strNum.find(strK) + 1
    else:
        return strNum.find(strK)

if문으로 간단하게 해결하였다. 아래에는 한 줄에 해결한 사람들의 코드다.

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

해석을 하자면 문자열로 캐스팅한 k가 문자열로 캐스팅한 num에 없다면 -1을 리턴하고 아니라면 문자열로 캐스팅한 num에서 문자열로 캐스팅한 k를 찾아 인덱스 번호로 값을 낸 다음 +1을 해서 리턴해주세요!

 

---

 

JavaScript

자바스크립트에는 indexOf()라는 함수가 있다.

function solution(num, k) {
    if (String(num).indexOf(String(k)) != -1) return String(num).indexOf(String(k)) + 1
    else return -1
}

방식은 파이썬과 같다. 아래에는 한 줄에 해결한 사람들의 코드다.

function solution(num, k) {
    return num.toString().split("").map((el) => Number(el)).indexOf(k) + 1 || -1
}

해석을 하자면 num이라는 매개변수를 .toString()으로 문자열화 시킨다.

.split("")으로 문자열화 시킨 num을 배열화한다. num이 만약 123이라면 ["1", "2", "3"] 이런식으로 만들어 줌

.map()으로 배열 각각의 요소를 숫자로 캐스팅해준다.  

.indexOf(k)로 인덱스 번호를 찾아내서 + 1을 더해 반환

|| 또는 -1 반환.

 

대단한 거 같다.

 

---

 

Java

자바에도 indexOf()라는 메소드가 있다.

class Solution {
    public int solution(int num, int k) {
        String sNum = String.valueOf(num);
        String sK = String.valueOf(k);
        
        if(sNum.indexOf(sK) != -1) return sNum.indexOf(sK) + 1;
            else return -1;
    }
}

아래에는 한 줄로 해결한 사람들의 코드이다.

class Solution {
   public int solution(int num, int k) {
      return ("-" + num).indexOf(String.valueOf(k));
   }
}

해석을 하자면 num 앞에 -이라는 문자열을 하나 더해서 (num은 알아서 문자열로 캐스팅됌), k라는 문자열을 찾아 인덱스 값으로 반환. - 때문에 + 1을 할 필요가 없음.

이 코드는 정말 천재 같다.