[LV.3] 연습문제 > 크기가 작은 부분문자열

문제

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

프로그래머스

---

문제 설명

• 숫자로 이루어진 문자열 t와 p가 주어질 때, t에서 p와 길이가 같은 부분문자열 중에서,
  이 부분문자열이 나타내는 수가 p가 나타내는 수보다 작거나 같은 것이 나오는 횟수를 반환
• 예를 들어, 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으로 시작하지 않음

---

첫 번째 시도 (성공. 개선 필요)

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

---

개선점

• 지금 코드에는 문자열 비교를 사용하고 있어서 숫자 크기 비교에서 문제가 생길 수 있음
  • 문자열 비교 대신 정수 비교를 사용
• len(p)는 여러 번 계산하지 말고 한 번만 변수로 저장하면 성능과 가독성에 좋음
• int(p)를 쓰게 되면, for 루프가 돌 때마다 매번 계산됨
  • 불필요한 중복 계산이므로, 한 번만 변환해서 변수에 저장하는 게 성능과 가독성에 좋음

 

---

개선된 코드

def solution(t, p):
    answer = 0
    p_len = len(p)
    p_int = int(p)

    for i in range(len(t) - p_len + 1):
        text = t[i:i + p_len]
        if int(text) <= p_int:
            answer += 1

    return answer

---

코드 설명

def solution(t, p):
    answer = 0
    p_len = len(p) # p와 같은 길이의 부분 문자열을 만들어야 하므로, 미리 길이를 저장
    p_int = int(p) # 비교를 위해 p를 숫자로 변환해 저장

    for i in range(len(t) - p_len + 1): # t에서 p_len 길이만큼 자를 수 있는 시작 인덱스 범위를 구함
        text = t[i:i + p_len]           # 현재 위치부터 p_len 길이만큼 잘라냄
        if int(text) <= p_int:          # 부분 문자열을 숫자로 바꿔 p_int와 비교
            answer += 1                 # 조건을 만족하면 카운트

    return answer  # 결과 반환