[Swift] - 백준 1543 : 문서 검색

뭔가 풀이가 굉장히 다양하게 나올 것 같은 문제였다

Swift가 문자열 다루기 까다로우니까 .. 냅다 배열로 받아 풀었다는 🐥

 

문제

세준이는 영어로만 이루어진 어떤 문서를 검색하는 함수를 만들려고 한다. 이 함수는 어떤 단어가 총 몇 번 등장하는지 세려고 한다. 그러나, 세준이의 함수는 중복되어 세는 것은 빼고 세야 한다. 예를 들어, 문서가 abababa이고, 그리고 찾으려는 단어가 ababa라면, 세준이의 이 함수는 이 단어를 0번부터 찾을 수 있고, 2번부터도 찾을 수 있다. 그러나 동시에 셀 수는 없다.
세준이는 문서와 검색하려는 단어가 주어졌을 때, 그 단어가 최대 몇 번 중복되지 않게 등장하는지 구하는 프로그램을 작성하시오.

 

입력

첫째 줄에 문서가 주어진다. 문서의 길이는 최대 2500이다. 둘째 줄에 검색하고 싶은 단어가 주어진다. 이 길이는 최대 50이다. 문서와 단어는 알파벳 소문자와 공백으로 이루어져 있다.

 

 

 

 

 

발로 그린 그림 ..

 

이렇게 두 배열을 나란히 두었다고 생각하고 word를 text의 첫인덱스부터 한칸 한칸 옮기는 방식으로 구현했다.

 

 

 

 

코드

var text = Array(readLine()!)
var word = Array(readLine()!)

var startIdx = 0
let wordCount = word.count
var count = 0

// 문서의 마지막 인덱스를 넘지 않을 때까지
while startIdx + wordCount <= text.count {
    
    if word.elementsEqual(text[startIdx..<startIdx + wordCount]) {
        
        count += 1
        // 같으면 시작점을 시작점 + word.count 으로변경
        startIdx += wordCount
        
    } else {
        
        // 같지 않으면 시작점을 시작점 + 1
        startIdx += 1
    }
}

print(count)

 

 

결과