[백준] java 11478 서로 다른 부분 문자열

https://www.acmicpc.net/problem/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의 모든 경우의 수의 조합을 구하면 될 것 같다. ababc로 예를 들면 a, ab, aba, abab, ababc 이렇게 말이다.
이렇게도 좋지만 나는 반복문의 조합 자리수를 한자리씩 올려가며 조합을 구하였다. 처음엔 a, b, a, b, c. 두 번째는 ab, ba, ab, bc. 세 번째는 세자리수이므로 aba, bab, abc. 코드 구현은 다음과 같다.

import java.io.BufferedReader;
import java.io.InputStreamReader;
import java.util.HashSet;
import java.util.Set;

public class Main {
	public static void main(String[] args) throws Exception{
		BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
		int num = 1;
		String s = br.readLine();
		Set<String> set = new HashSet<>();
		while( num != s.length() ) {		
			for( int i = 0 ; i < s.length()-num+1; i++ ) {
				set.add( s.substring( i, num + i) );
			}
			num++;
		}
		set.add(s);
		System.out.println( set.size() );
	}
}

먼저 중복 값이 있을 수 있으므로 HashSet을 사용하였다. 이후 자리수가 저장된 num을 선언하고 반복문을 돌렸다. 자리수만큼 subString을 사용하면 문자열이 알맞게 잘려 add된다.

나가는 글

집합과 맵 카테고리도 이렇게 끝났다. 이번 카테고리는 다른 곳과 비교해서 난이도가 어려운 문제가 없던 것 같다.