문제

문자열 S가 주어졌을 때, S의 서로 다른 부분 문자열의 개수를 구하는 프로그램을 작성하시오.

부분 문자열은 S에서 연속된 일부분을 말하며, 길이가 1보다 크거나 같아야 한다.

예를 들어, ababc의 부분 문자열은 a, b, a, b, c, ab, ba, ab, bc, aba, bab, abc, abab, babc, ababc가 있고, 서로 다른것의 개수는 12개이다.

코드

inpt = input()

container = set()
ans =1
while ans < len(inpt):
    for i in range(len(inpt)):
        container.add(inpt[i:i+ans])
    ans +=1

print(len(container)+1)

설명

중복이 제거된 부분집합을 구해야 하기 때문에 set을 써서 길이가 1인 문자들부터 길이가 주어진 문자열의 길이만큼 분리해서 set에 저장한 후 set의 길이를 출력해주었다.