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

예제 입력 1

ababc

예제 출력 1

12

📝 풀어보기

📌 문자열을 입력받고 입력받은 문자열을 저장할 set을 생성한다.

문자열의 길이, i만큼 반복하면서 temp에 S의 i부터 j+1까지 슬라이싱한 값을 넣고 set에 temp값을 추가시킨다.

set은 중복이 없으므로 set의 길이가 서로 다른 부분 문자열의 개수다.

S = input()
set_S = set()

for i in range(len(S)):
    for j in range(i, len(S)):
        temp = S[i:j+1]
        set_S.add(temp)
print(len(set_S))