[kotlin] 크기가 작은 부분문자열 (프로그래머스 알고리즘) : substring(), windowed()

 

 

 

 

 

문제

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

 

 

 

---

 

성공 코드

 

class Solution {
    fun solution(t: String, p: String): Int {
        
        var count = 0
        // p의 길이 만큼, t의 문자열을 뽑아서 비교
        for (i in 0 .. t.length - p.length){
            val substring = t.substring(i, i + p.length) // i부터 시작하여 i + p.length 전까지 문자열 자르기
            
            if(substring.toLong() <= p.toLong()){
                count++
            }
        }
        return count
    }
}

 

---

 

설명

 

1. p의 길이만큼 t의 문자열을 뽑아서 비교하기 위해 for문을 만든다.
   for문은 0부터 (총 길이 - p의 길이)만큼 반복한다.
   (substring(i, i + p.length)를 수행하기 위해 이렇게 범위를 설정)
2. 뽑아낸 문자열마다 p와 크기를 비교한다.
   p보다 작거나 같으면 count를 하나씩 늘린다.
3. 모든 비교를 완료하면 count를 리턴한다.

 

 

💡 substring()

val str = "Hello, world!"

val result1 = str.substring(7) // "world!"
val result2 = str.substring(7, 12) // "world"
val result3 = str.substring(7..11) // "world"

• 문자열에서 특정 부분 문자열을 추출하는데 사용하는 메서드
• substring(startIndex: Int): String: 시작 인덱스부터 문자열의 끝까지 추출
• substring(startIndex: Int, endIndex: Int): String: 시작 인덱스부터 끝 인덱스까지 추출
• substring(range: IntRange): String: 지정된 범위에 해당하는 부분 문자열을 추출

 

 

---

 

다른 방법

 

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 }
    }
}

• map을 이용하여 새로운 배열을 바로 반환하여 count를 세서 간단하게 코드를 줄인 풀이다.

 

💡map

val numbers = listOf(1, 2, 3, 4, 5)
val squaredNumbers = numbers.map { it * it }
println(squaredNumbers) // 출력: [1, 4, 9, 16, 25]

• 각 요소를 주어진 람다 함수에 적용하여 새로운 컬렉션을 생성한다.
• 기존 컬렉션의 각 요소에 대해 주어진 람다 함수가 적용된 결과를 담은 새로운 컬렉션을 반환한다.

 

 

class Solution {
    fun solution(t: String, p: String): Int =
        t.windowed(p.length).count {
            it.toLong() <= p.toLong()
        }
}

• windowed()를 이용한 풀이다. 윈도우의 사이즈만 p.length로 설정되어 step은 기본값인 1로 설정되었다.

 

 

💡windowed()

fun <T> Iterable<T>.windowed(
    size: Int,
    step: Int = 1,
    partialWindows: Boolean = false
): List<List<T>>

val list = listOf(1, 2, 3, 4, 5, 6, 7, 8, 9, 10)

// 윈도우 크기가 3이고, 이동 거리가 1인 경우
val result1 = list.windowed(size = 3, step = 1)
// 결과: [[1, 2, 3], [2, 3, 4], [3, 4, 5], [4, 5, 6], [5, 6, 7], [6, 7, 8], [7, 8, 9], [8, 9, 10]]

// 윈도우 크기가 4이고, 이동 거리가 2인 경우
val result2 = list.windowed(size = 4, step = 2)
// 결과: [[1, 2, 3, 4], [3, 4, 5, 6], [5, 6, 7, 8], [7, 8, 9, 10]]

// 윈도우 크기가 2이고, 이동 거리가 3인 경우
val result3 = list.windowed(size = 2, step = 3)
// 결과: [[1, 2], [4, 5], [7, 8], [10]]

// 부분 윈도우 포함
val result4 = list.windowed(size = 3, step = 2, partialWindows = true)
// 결과: [[1, 2, 3], [3, 4, 5], [5, 6, 7], [7, 8, 9], [9, 10]]

// 윈도우 크기만 설정, 이동 거리는 기본 1로 설정됨
val result5 = list.windowed(3)
// 결과 : [[1, 2, 3], [2, 3, 4], [3, 4, 5], [4, 5, 6], [5, 6, 7], [6, 7, 8], [7, 8, 9], [8, 9, 10]]

• 주어진 크기의 창을 이동하면서 컬렉션을 분할하는데 사용
• 반환 값은 주어진 크기의 부분 컬렉션 또는 윈도우
• size: 생성할 각 윈도우의 크기를 나타내는 값
• step: 윈도우를 이동시키는 간격을 나타내는 값. 기본값은 1이며, 이는 한 번에 한 요소씩 이동함을 의미
• partialWindows: 마지막 부분 윈도우의 크기가 size보다 작을 때, 이 부분 윈도우를 포함할지 여부를 결정하는 불리언 값. 기본값은 false이며, 이 경우 부분 윈도우는 포함되지 않음

 

 

 

정확히 기억은 안나지만, 이 이외에도 팀원분이 공유해주신 코드 중에 chucked를 이용하신 코드가 있었다.
결국엔 스트링 값으로 p와 비교하게 되면서 toInt()나 toLong()과 같은 형변환을 신경쓸 필요가 없는 코드였다.