NIRVANA

[level 1] 크기가 작은 부분 문자열 본문

Coding test(Python3)/Programmers

[level 1] 크기가 작은 부분 문자열

녜잉 2023. 7. 25. 17:50

문제

숫자로 이루어진 문자열 t와 p가 주어질 때, t에서 p와 길이가 같은 부분문자열 중에서, 이 부분문자열이 나타내는 수가 p가 나타내는 수보다 작거나 같은 것이 나오는 횟수를 return하는 함수 solution을 완성하세요.

예를 들어, t="3141592"이고 p="271" 인 경우, t의 길이가 3인 부분 문자열은 314, 141, 415, 159, 592입니다. 이 문자열이 나타내는 수 중 271보다 작거나 같은 수는 141, 159 2개 입니다.

 

문제 풀이 접근법1

1) len(t) - len(p) +1번 반복(파이썬은 -1이니까 +2로 해주었다.)하며 배열 슬라이싱을 상용해 i부터 i+len(p) 만큼 문자열 t를 슬라이싱 한다. 

*range니까 0부터 시작해서 사실 +1해도 상관 없다!

2) 문자열의 길이가 p와 같은 t의 부분 문자열을 int() 함수를 사용하여 숫자로 변경한 뒤, p와 비교한다. 

3) p보다 작다면 answer에 +1을 한다.

4) answer를 반환한다.

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

배열 슬라이싱을 했을 때, 숫자가 모자라면 그냥 끝까지만 해서 n만큼 되지 않아서 문자열의 길이가 p와 같은지 확인하는 코드를 따로 추가했다. 

그나저나 이번 문제 테스트 케이스 수 왤케 많아!! 놀랐다.