[BOJ] #11478_서로 다른 부분 문자열의 개수(Python3, 파이썬)

본 게시글은 백준 저지 알고리즘 문제를 '파이썬, Python3' 언어로 풀이한 내용을 주관적으로 정리하였으며, 내용과 관련된 코드리뷰 및 피드백 환영합니다.

티어

Silver III

문제

문자열 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

12

---

풀이

import sys
input = sys.stdin.readline

s = input().rstrip()
total = set()
for i in range(len(s)):
    for j in range(len(s)):
        total.add(s[i:j+1]) #i번째 문자부터 부분문자열 구하기

print(len(total)-1)

부분문자열은 문자열의 슬라이싱을 이용해 풀어주면 된다. 중복을 제거하는 데에 있어서 가장 좋은 자료형은 바로 집합이다. 집합에 부분문자열을 모두 넣는데, 이때 중복은 자동으로 제거해주기 때문에, 집합의 길이 즉, 개수를 출력해주면 된다.