[Java&알고리즘] 크기가 작은 부분 문자열 (Level 1)

 

 

환경

언어 : 자바

단계 : 레벨 1단계 (입문자용)

정답률 : 68%

 

 

 

 

제목 

 

크기가 작은 부분 문자열

 

 

 

 

 

문제 설명 및 코드

 

 

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

 

 

 

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

 

 

 

 

 

입출력 예

 

 

 

 

입출력 예 설명

 

입출력 예 #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;
        for(int a=0; a<=t.length()-p.length(); a++){        
            String refStr = t.substring(a,a+p.length()); 
            if(Long.parseLong(refStr) <= Long.parseLong(p)){
                ++answer;
            } 
        }
        return answer;
    }
}

 

 

우선 반복문의 길이를 t의 길이에서 p의 길이를 뺀 만큼만 설정해줍니다.

 

서브스트링을 사용하기 위해서인데 그냥 t의 길이를 설정해주면 아래와 같이 StringIndexOutOfBoundsException 에러가 발생하게 됩니다.

 

 

그리고 p의 길이에 맞게 서브스트링을 설정하여 refStr에 넣어준다음

 

문자열을 Long 타입으로 변환하여 비교를 해준 후 조건에 맞다면 answer를 증가시켜줍니다.

 

Long타입이 아닌 Int형으로 비교할 경우 아래와 같이 런타임 에러가 발생하니 주의가 필요합니다.

 

 

 

 

 

 

 

 

 

 

 

다른사람 풀이

 

 

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();
    }
}

 

 

이 코드는 스트림으로 해결한 코드입니다.

 

LongStream의 range로 반복하면서

 

서브스트링을 사용한 다음

 

롱 타입으로 변환하여

 

filter로 조건을 비교한 다음, 맞다면 count를 해줍니다.

 

스트림의 세계는 봐도봐도 신기한 것 같습니다...