BOJ-11478 서로 다른 부분 문자열의 개수 Swift

문제

문자열 S가 주어졌을 때, S의 서로 다른 부분 문자열의 개수를 구하는 프로그램을 작성하시오.

부분 문자열은 S에서 연속된 일부분을 말하며, 길이가 1보다 크거나 같아야 한다.

예를 들어, ababc의 부분 문자열은 a, b, a, b, c, ab, ba, ab, bc, aba, bab, abc, abab, babc, ababc가 있고, 서로 다른것의 개수는 12개이다.

입력

첫째 줄에 문자열 S가 주어진다. S는 알파벳 소문자로만 이루어져 있고, 길이는 1,000 이하이다.

출력

첫째 줄에 S의 서로 다른 부분 문자열의 개수를 출력한다.

풀이

1. 문자의 부분문자열을 Set에 저장

2. Set.count 출력

 

import Foundation

var strArr = readLine()!.map{String($0)}
var strSet = Set<String>()

for i in 0..<strArr.count {
    var str = ""
    for j in i..<strArr.count {
        str = str + strArr[j]
        strSet.insert(str)
    }
}
print(strSet.count)

 

 

 

 

<시간초과>

import Foundation

var str = readLine()!.map{String($0)}
var strSet = Set<String>()
var length = 0

for i in 0..<str.count {
    var count = str.count - i
    for j in 0..<count {
        let str = str[j...j+length].joined(separator: "")
        strSet.insert(str)
    }
    length += 1
}
print(strSet.count)

 

처음 시간초과가 났을때 부분문자열을 만드는 과정이 문제라 생각했었다

부분문자열의 길이를 1씩 증가시켜 각 인덱스마다 부분문자열을 추출하는 방식을 사용했다가,

더 간단하게 부분문자열을 추출하는 방법이 있어서 그 방법을 사용했지만

둘 다 2중 for문을 사용해서 결국 O(n^2) 라 생각한다

아마 문제는 배열에서 범위 추출하는 과정이 문제라 생각이 든다

배열에 대해서 따로 정리를 해봐야겠다