02/13 알고리즘 공부(43) - 크기가 작은 부분문자열

조건

숫자로 이루어진 문자열 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. 반복문을 이용해 p의 길이만큼 t를 자른 문자열을 만든다.
2. p와 비교해 문자열이 작으면 answer에 1을 더한다.

 

초기 코드

class Solution {
    fun solution(t: String, p: String): Int {
        var answer: Int = 0
        return answer
    }
}

 

내 코드

class Solution {
    fun solution(t: String, p: String): Int {
        var answer: Int = 0
        for(i in 0..t.length-p.length){
            if(t.slice(i .. i+p.length-1) <= p) answer ++
        }
        return answer
    }
}

slice를 이용해 자른 문자열을 이용해 비교해 보았다. map을 이용해 콜렉션을 만든다음 size나 count를 이용하는 방법도 될 것 같다.

다른 사람의 풀이

class Solution {
    fun solution(t: String, p: String): Int {
        return (0..t.length - p.length)
            .map{ t.substring(it until it + p.length) }
            .count { it <= p }
    }
}

 

개선점 또는 배운점

1. 예상했던 풀이가 나왔다. map을 이용해  p와 길이가 같은 string을 원소로 가지는 콜렉션을 만든 다음 count를 이용해 조건을 만족하는 원소를 셌다.
2. for문과 if문 등 기본 문법을 이용하는 방법이 익숙해 다른 메소드를 사용하는 것이 어색하다. 의식해서라도 메소드를 사용해서 풀어야겠다.

 

개선된 코드

class Solution {
    fun solution(t: String, p: String): Int = (0..t.length-p.length).map{t.slice(it until it+p.length)}.count{it <= p}  
}