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

프로그래머스 LV.1 모음

크기가 작은 부분문자열

문제 설명

숫자로 이루어진 문자열 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

입출력 예 설명

입출력 예 #1

• 본문과 같습니다.

입출력 예 #2

• p의 길이가 1이므로 t의 부분문자열은 "5", "0", 0", "2", "2", "0", "8", "3", "9", "8", "7", "8"이며 이중 7보다 작거나 같은 숫자는 "5", "0", "0", "2", "2", "0", "3", "7" 이렇게 8개가 있습니다.

입출력 예 #3

• p의 길이가 2이므로 t의 부분문자열은 "10", "02", "20", "03"이며, 이중 15보다 작거나 같은 숫자는 "10", "02", "03" 이렇게 3개입니다. "02"와 "03"은 각각 2, 3에 해당한다는 점에 주의하세요

---

제출

import Foundation

func solution(_ t:String, _ p:String) -> Int {
    var cnt = 0
    var n = Int(p)!
    for i in 0...t.count - p.count{
        cnt += Int(t.map{String($0)}[i..<i+p.count].joined())! <= n ? 1 : 0
    }
    return cnt
}

최종 결과 값을 담을 변수 cnt와 입력받은 문자열 p를 Int형으로 변환해 저장한 변수 n을 선언한다.
i를 0부터 (t의 길이 - p의 길이)까지 1씩 증가시키며 반복한다. (반복문에서 i에 p의 길이를 더하기 때문에)
t문자열을 배열로 바꾼 후, 인덱스 i부터  i+p까지의 문자열을 Int로 바꿨을 때, n 보다 작다면 cnt 변수에 1을 아니라면 0을 더한다.

다른 풀이

import Foundation

func solution(_ t:String, _ p:String) -> Int {
    let n = p.count, p_num = Int(p)!
    var result = 0
    
    for i in 0...t.count-n{
        let s = t.dropFirst(i).prefix(n)
        if p_num >= Int(s)!{result += 1}
    }
    return result
}

n은 p의 길이를, p_num은 문자열 p를 Int형으로 변환하여 저장한다.
최종 결과 값을 담을 변수 result도 선언한다.

i를 0부터 (t의 길이 - n)까지 1씩 증가시키며 반복한다.
문자열 t의 i번째 문자부터 n개의 문자열을 s에 저장한다.
문자열 s를 Int로 변환한 값이 p_num보다 작다면, result를 1 증가시킨다.