99클럽 코테 스터디 2일차 TIL - 크기가 작은 부분문자열

오늘의 문제

프로그래머스 - 크기가 작은 부분문자열

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

문제내용과 풀이

문제 설명

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

입출력 예

 "3141592" - "271" - 2
"500220839878" - "7" - 8
"10203" - "15" - 3

 

이번 문제의 주제는 부분문자열이었고, substring 에 대한 내용임을 알 수 있었다.

 

1.처음 풀이

class Solution {
    public int solution(String t, String p) {
        long pValue = Long.parseLong(p);
        int count = 0;
        
        for (int i = 0; i <= t.length() - p.length(); i++) {
            String substrT = t.substring(i, i + p.length());
            long tsValue = Long.parseLong(substrT);
            if (tsValue <= pValue) {
                count++;
            }
        }
        
        return count;
    }
}

 

String 메서드인 substring을 이용해서 t.length()에서 p.length()를 뺀 length 까지 반복해서 해결 할 수 있었다.

평소대로 반복문을 만들며 처음엔 i의 범위를 < 로 놓았는데, 정답이 되지 않아 계속 생각해보니

i가 0부터 시작하더라도 p의 length는 0부터 시작하지 않은 길이라 부등호를 <=로 놓아야 한다는 걸 깨달았다...

t.length() - p.length() 같은 부분을 변수로 담을까 했지만 크게 차이는 없을 것 같다고 생각했다.

 

2. 다른 풀이

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

 

비슷하지만 substring 할 때 부터 변수없이 바로 비교를 해서 증가시키는 방법이 있었다.

코드는 좀 더 간단해졌지만, 처음 보면 알아보기 힘들 것 같다.

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

        for(int i=0;i<t.length()-p.length()+1;i++){

            if(Long.compare(Long.parseLong(t.substring(i,i+p.length())),
                              Long.parseLong(p)) != 1){
                answer += 1;
            }
        }
        return answer;
    }
}

 

다른 많은 풀이가 있었는데, compare 를 사용한 풀이도 있었다. 

compare를 통해 두 타입의 값을 비교해서 compare(x , y) 인 경우 x가 y보다 작으면 음수, 같으면 0, 크면 양수를 반환해서count 하는 방법이 있었다.

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

 

람다로 해결한 사람도 있었는데, 그냥 for문 돌린 것 보다 나아 보이진 않았지만 이런 방법도 있구나.. 했다.