[BOJ] 11478 서로 다른 부분 문자열의 개수 (Week 7, No.10)

문제

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

예제 입력 1

ababc

예제 출력 1

12

---

풀이

길이가 1부터 s의 길이까지 문자열을 자른다.

자른 문자열을 hashset에 넣어 중복을 확인한다.

hashset에 자른 문자열이 존재하지 않다면 set에 넣는다.
존재한다면 다음 문자열로 넘어간다.

마지막에는 set의 크기를 반환한다.
set에는 중복되지 않은 자른 문자열만 들어있어서 set의 크기가 겹치지 않는 다양한 문자열의 수가 된다.

시간복잡도: O(N*(N+1)/2) = O(N^2) - 1부터 n까지의 수를 더한 것
공간복잡도: O(M) - M은 겹치지 않는 문자열의 개수

---

코드

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

public class Main {
    public static void main(String[] args) throws IOException {
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        String s = br.readLine();

        // base case
        if (s.length() == 0) {
            System.out.println(0);
            return;
        }

        Set<String> set = new HashSet<>();
        int left, right;
        String subString = "";
        
        for (int i = 0; i < s.length(); i++) {
            left = 0;
            right = left + i;
            while (right < s.length()) {
                if (right == s.length() - 1) {
                    subString = s.substring(left);
                } else {
                    subString = s.substring(left, right + 1);
                }
                if (!set.contains(subString)) {
                    set.add(subString);
                }
                left++;
                right++;
            }
        }
        System.out.println(set.size());
    }
}