[Baekjoon/C++] 11478번 - 서로 다른 부분 문자열의 개수

[백준] Baekjoon Online Judge

문제로 이동

 

문제

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

 

---

예제 입력	예제 출력
ababc	12

---

풀이

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

int main() {
	ios_base::sync_with_stdio(false);
	cin.tie(NULL);

	unordered_set <string> set;
	string s;
	cin >> s;

	int cnt = 0;

	for (int i = 1; i <= s.length(); i++) {
		for (int j = 0; j <= s.length() - i; j++)
			set.insert(s.substr(j, i));

		cnt += set.size();
		set.clear();
	}

	printf("%d\n", cnt);

	return 0;
}

시간 단축을 위해 정렬을 하지 않는 unordered_set을 사용해서 문제를 풀었다. set에 문자열은 substr()로 쪼개서 넣었다. 길이가 1인 문자열을 모두 set에 넣고 나면 그 수를 cnt에 저장하고, set을 초기화한 뒤 길이가 2인 문자열을 set에 넣는 식으로 코드를 구현해서 메모리 절약과 시간 단축을 했다.