백준 11478 서로 다른 부분 문자열의 개수(c++)

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

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

<코드>

#include <iostream>
#include <set>
#include <string>
#include <vector>
#include <functional>
#include <algorithm>

using namespace std;

int main()
{
	string str;
	set<string>s;
	vector<string>v;
	cin>>str;
	for(int i=0;i<str.size();i++)
	{
		string ans="";
		for(int j=i;j<str.size();j++){
			ans += str[j];
			v.push_back(ans);
		}
		v.push_back(ans);
	}
	sort(v.begin(),v.end());
	v.erase(unique(v.begin(),v.end()),v.end());
	
	cout<<v.size();
}

<풀이과정>

각각의 부분 문자열을 벡터에 추가한다음, 중복된 부분을 제거해줘야 한다.

unique함수와 erase를 이용해 바로 제거할 수 있다.

sort(v.begin(),v.end()) =>각각의 원소 오름차순 정렬

v.erase(unique(v.begin(),v.end()),v.end())=>오름차순 정렬된 원소들 중 중복되는 원소 제거