[프로그래머스] lv.1 크기가 작은 부분문자열 python

https://school.programmers.co.kr/learn/courses/30/lessons/147355

프로그래머스

문제 설명

숫자로 이루어진 문자열 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으로 시작하지 않습니다.

 

문제 풀이

p의 길이를 idx에 담고

idx만큼씩 t를 slice해서 새로운 배열 arr에 담아준다.

arr에 담겨있는 문자들을 int형으로 변환해서 p와 비교해주고

p보다 작은 수일 경우 cnt를 1씩 더해준다.

 

cnt를 answer에 담아주고 answer를 리턴해주면 끝 !

 

def solution(t, p):
    idx=len(p)
    arr=[]
    cnt=0
    for i in range(len(t)-idx+1):
        arr.append(t[i:i+idx])
        
    for i in range(len(arr)):
        if int(arr[i])<=int(p):
            cnt+=1

    answer = cnt
    return answer