BOJ: #1543 - 문서 검색

문자열에 인덱스로 접근하기 위에 subscript를 익스텐션으로 구현했다. 혹은 문자열을 배열로 변환해서 풀어도 된다. 나는 두 개의 커서(문자열 커서, 패턴 커서)를 이용해 풀었다.

문제

https://www.acmicpc.net/problem/1543

내가 작성한 코드

Swift

extension String {
    subscript(offset: Int) -> String {
        get {
            let index = String.Index(utf16Offset: offset, in: self)
            return String(self[index])
        }
    }
}

let str = readLine()!
let pattern = readLine()!
let n = str.count
let m = pattern.count
var cur = 0 // str cursor
var answer = 0
while cur < n {
    var i = 0 // pattern cursor
    if str[cur] == pattern[i] {
        var isFind = true
        while i < m {
            if cur + i >= n {
                isFind = false
                break
            }
            if str[cur + i] != pattern[i] {
                isFind = false
                break
            }
            i += 1
        }
        if isFind {
            cur += i
            answer += 1
        } else {
            cur += 1
        }
    } else {
        cur += 1
    }
}
print(answer)