Algorithm/프로그래머스

프로그래머스 - 크기가 작은 부분문자열 (Swift, Level1)

Brad_Heo 2023. 2. 14. 16:01

문제 👇

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

문제설명 📝

숫자로 이루어진 문자열 t와 p가 주어질 때, t에서 p와 길이가 같은 부분문자열 중에서, 이 부분문자열이 나타내는 수가 p가 나타내는 수보다 작거나 같은 것이 나오는 횟수를 return하는 함수 solution을 완성하세요.

예를 들어, t="3141592"이고 p="271" 인 경우, t의 길이가 3인 부분 문자열은 314, 141, 415, 159, 592입니다. 이 문자열이 나타내는 수 중 271보다 작거나 같은 수는 141, 159 2개 입니다.

문제풀이 ✅

  1. t, p의 길이 가져오기
  2. String -> Array로 변환(p의 값과 대조를 위해)

문제풀이 답

func solution(_ t:String, _ p:String) -> Int {

    let tCount = t.count
    let pCount = p.count

    var store: [String] = []
    var count = 0
    var result = 0

    t.forEach { element in
        if store.count == tCount - pCount + 1 {
            return
        } else {
            store.append(String(Array(t)[count...count + pCount - 1]))
            count += 1
        }
    }

    store.forEach { element in
        if element <= p {
            result += 1
        }
    }
    return result
}