[백준 알고리즘] 11478번: 서로 다른 부분 문자열

11478번: 서로 다른 부분 문자열

문자열 S가 주어졌을 때, S의 서로 다른 부분 문자열의 개수를 구하는 프로그램을 작성하시오.
부분 문자열은 S에서 연속된 일부분을 말하며, 길이가 1보다 크거나 같아야 한다. 예를 들어, ababc의 부분 문자열은 a, b, a, b, c, ab, ba, ab, bc, aba, bab, abc, abab, babc, ababc가 있고, 서로 다른것의 개수는 12개이다.

접근 방법:

일단 중복된 문자열을 허용하지 않기 때문에 set자료구조를 사용하기로 했다. 그리고 string 라이브러리에 포함된 substr을 사용하여 첫 글자부터 그 다음 글자들을 부분 문자열로 가져올 문자열 길이를 1씩 늘려가며 모두 가져와 set에 넣어주었다. 이렇게 하면 가능한 모든 부분 문자열이 중복없이 set에 들어가고 최종적으로는 set의 사이즈를 출력하면 될 것이다.

통과 코드:

#include <set>
#include <string>
#include <iostream>

using namespace std;

int main(){
    string s;
    set<string> strset;
    cin >> s;

    set<string>:: iterator iter;

    for (int i = 0 ; i < s.size() ; i++){
        for (int j = 1 ; j <= s.size()-i ; j++){
            string temp = s.substr(i, j);
            strset.insert(temp);
        }
    }
    cout << strset.size() << endl;
    return 0;
}

Written by@전여훈 (Click Me!)
고민이 담긴 코드를 만들자, 고민하기 위해 공부하자.