[백준/C++] 11478번: 서로 다른 부분 문자열의 개수

문제

문자열 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의 서로 다른 부분 문자열의 개수를 출력한다.

풀이

"서로 다른" 부분 문자열을 찾는 문제로, set을 쓰면 간편하게 풀 수 있다. 

처음에는 개수 별로 구분해서 풀었는데, 시작 인덱스로도 구분할 수 있어서 이걸로 풀었다. 전자보다 조금 더 간단한 것 같다. 

#include <iostream>
#include <set>
using namespace std;

int main() {
    string s;

    cin >> s;

    set<string> subStr;
    /* 1) [0] [1] [2] [3] [4]
     * 2) [0-1] [1-2] [2-3] [3-4]
     * 3) [0-1-2] [1-2-3] [2-3-4]
     * 4) [0-1-2-3] [1-2-3-4]
     * 5) [0-1-2-3-4]
     * => 0으로 시작하는 것 / 1로 시작하는 것 / 2로 시작하는 것 / ...
     */
    for (int i=0; i<s.length(); i++){
        string tmp = "";
        for (int j=i; j<s.length(); j++){
            tmp += s[j];
            subStr.insert(tmp);
        }
    }

    cout << subStr.size();
}