[프로그래머스 스쿨] 코딩테스트 연습 기초 - 크기가 작은 부분 문자열

김현정·2025년 4월 18일

프로그래머스 스쿨 : 코딩테스트 연습

목록 보기

34/45

문제 설명

숫자로 이루어진 문자열 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에 해당한다는 점에 주의하세요

기본 코드

class Solution {
    public int solution(String t, String p) {
        int answer = 0;
        return answer;
    }
}

내가 구현한 코드

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

다른사람 푼 코드

import java.util.stream.LongStream;

class Solution {
    public int solution(String t, String p) {
        long targetNumber = Long.parseLong(p);
        int targetNumberLength = p.length();

        return (int) LongStream.range(0L, t.length() - targetNumberLength + 1L)
                .mapToObj(i -> t.substring((int) i, (int) i + targetNumberLength))
                .mapToLong(Long::parseLong)
                .filter(number -> number <= targetNumber)
                .count();
    }
}

문제 해결

반복문을 통해 문자열 t를 문자열 p의 길이만큼 잘라 p보다 작은 부분문자열 t의 개수를 구하는 문제다.
문자열 길이가 10,000까지 존재할 수 있으므로 정수형으로 변환 시 long 타입으로 변환해야 한다.
반복문의 인덱스가 넘기지 않으면서 문자열을 자르기 위해 문자열 t의 길이에서 문자열 p의 길이만큼 뺀 숫자만큼 반복한다. substring을 통해 t를 i부터 i+p.length() 만큼 문자열을 잘라 temp에 담는다.
부분문자열 temp의 값이 p보다 작은 경우 answer를 증가시킨다.