프로그래머스 Lv.1 Swift 크기가 작은 부분 문자열

 

Swift에서는 다른 언어에서처럼 String형에서 Index에 Int로 접근할 수 없다. String 인덱스 값에 Int가 아닌 인덱스 자료형이 위치해 있어(Swift에서 String은 Reference Typed이 아닌 Value Type이기 때문이라고 한다.)  Substring 사용이 꽤 번거롭고 귀찮기는 하다.

 

문제

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

 

입출력 예

t	p	result
"3141592"	"271"	2
"500220839878"	"7"	8
"10203"	"15"	3

 

 

 

풀이

뭔가 더 좋은 방법이 있을 것 같은데 일단 Swift에서는 Python이나 다른 언어처럼 String[Int형]으로 부분문자열을 구할 수 없기 때문에 String.Index를 이용해서 풀어보았다.

 

func solution(_ t:String, _ p:String) -> Int {
    var counter: Int = 0
    for str in 0...(t.count - p.count) {
        let startIndex = t.index(t.startIndex, offsetBy: str)
        let index = t.index(startIndex, offsetBy: p.count)
        let sub = t[startIndex..<index]

        if Int(sub) ?? 0 <= Int(p) ?? 0 {
            counter += 1
        }
    }    
    return counter
}

 

 

 

참고

https://jeong9216.tistory.com/556

[Swift] String.Index 활용법

https://jcsoohwancho.github.io/2019-11-19-Swift-String-%ED%9A%A8%EC%9C%A8%EC%A0%81%EC%9C%BC%EB%A1%9C-%EC%93%B0%EA%B8%B0/