[프로그래머스]크기가 작은 부분문자열 - python, java

`크기가 작은 부분문자열`

문제 설명

• 숫자로 이루어진 문자열 t와 p가 주어질 때, t에서 p와 길이가 같은 부분문자열 중에서, 이 부분문자열이 나타내는 수가 p가 나타내는 수보다 작거나 같은 것이 나오는 횟수를 return하는 함수 solution을 완성하세요.
• 예를 들어, t="3141592"이고 p="271" 인 경우, t의 길이가 3인 부분 문자열은 314, 141, 415, 159, 592입니다. 이 문자열이 나타내는 수 중 271보다 작거나 같은 수는 141, 159 2개 입니다.

제한사항

• 1 ≤ p의 길이 ≤ 18
• p의 길이 ≤ t의 길이 ≤ 10,000
• t와 p는 숫자로만 이루어진 문자열이며, 0으로 시작하지 않습니다.

입출력 예

---

입출력 예 #1
본문과 같습니다.

입출력 예 #2

• p의 길이가 1이므로 t의 부분문자열은 "5", "0", 0", "2", "2", "0", "8", "3", "9", "8", "7", "8"이며 이중 7보다 작거나 같은 숫자는 "5", "0", "0", "2", "2", "0", "3", "7" 이렇게 8개가 있습니다.

 

입출력 예 #3

• p의 길이가 2이므로 t의 부분문자열은 "10", "02", "20", "03"이며, 이중 15보다 작거나 같은 숫자는 "10", "02", "03" 이렇게 3개입니다. "02"와 "03"은 각각 2, 3에 해당한다는 점에 주의하세요

---

java code

public class Solution {
    public int solution(String t, String p) {
        int answer = 0;
        Long longP = Long.parseLong(p);
        for(int i=0; i < t.length()-p.length()+1; i++){
            Long longLen = Long.parseLong(t.substring(i,i+p.length()));
            if(longLen <= longP){
                answer++;
            }
        }
        return answer;
    }
}

1. cnt 변수를 0으로 초기화합니다.
2. 문자열 p를 Long 타입으로 변환합니다.
3. t 문자열에서 p의 길이만큼 뒤쪽으로 이동하면서 substring을 추출합니다. 추출할 수 있는 substring의 개수는 t의 길이에서 p의 길이를 뺀 값과 같습니다.
4. 추출한 substring을 Long 타입으로 변환합니다.
5. index가 lp보다 작거나 같으면 cnt를 1 증가시킵니다.
6. cnt 값을 반환합니다.

 

python code

def solution(t, p):
    answer = 0
    long_p = int(p)
    for i in range(len(t) - len(p) + 1):
        sub_str = int(t[i:i+len(p)])
        if sub_str <= long_p:
            answer += 1
    return answer

코드는 주어진 문자열 t에서 길이가 p와 같은 부분문자열을 모두 찾아, 해당 부분문자열이 나타내는 숫자가 p가 나타내는 숫자보다 작거나 같은 경우를 count하는 방식으로 작성됩니다.

먼저 문자열 t에서 길이가 p와 같은 부분문자열을 찾기 위해서는 반복문을 사용합니다. 이때, 반복문의 종료조건은 t 문자열의 마지막 글자에서 p 문자열의 길이를 뺀 위치까지만 검사하는 것입니다. 이렇게 하는 이유는, t 문자열에서 길이가 p와 같은 부분문자열을 찾을 때, 마지막 부분문자열은 t 문자열의 마지막 위치에서 길이가 p인 문자열을 시작하는 경우이기 때문입니다.

반복문 안에서는 t 문자열에서 i 위치부터 길이가 p인 부분문자열을 추출하여 변수 sub_str에 저장합니다. 이후, sub_str이 나타내는 숫자가 p가 나타내는 숫자보다 작거나 같은 경우, count를 1 증가시킵니다. 숫자 비교를 위해 int() 함수를 사용하여 문자열을 숫자로 변환합니다.

최종적으로, count 값을 반환합니다.