[백준] 11478 : 서로 다른 부분 문자열의 개수 (Python/파이썬)

문제 링크

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

출력

12

---

정답 및 풀이

내가 푼 풀이

import sys

s = sys.stdin.readline().strip()

list_s = []
for i in range(len(s)):
    for j in range(i+1,len(s)+1):
        list_s.append(s[i:j])

set_s = set(list_s)
print(len(set_s))

1. s = sys.stdin.readline().strip()
   • 문자열 s를 입력받고 개행 문자를 제거한다.
2. list_s = []
   • 부분 문자열들을 저장할 리스트를 생성한다.
3. for i in range(len(s)):
   • 문자열의 첫 번째 문자부터 끝까지 순회한다.
4. for j in range(i+1, len(s)+1):
   • 시작 인덱스 i로부터 끝 인덱스를 j로 이동하며 부분 문자열을 생성한다.
5. list_s.append(s[i:j])
   • 문자열 슬라이싱을 통해 부분 문자열을 리스트에 추가한다.
6. set_s = set(list_s)
   • 리스트를 집합으로 변환하여 중복된 부분 문자열을 제거한다.
7. print(len(set_s))
   • 서로 다른 부분 문자열의 개수를 출력한다.

다른 풀이

import sys
s = sys.stdin.readline().strip()

set_s = set()

for i in range(len(s)):
    for j in range(i+1, len(s)+1):
        set_s.add(s[i:j])

print(len(set_s))

• 바로 set 에 추가

---