백준(baekjoon) 11478번: 서로 다른 부분 문자열의 개수(실버 3, 파이썬 PYTHON) - 집합과 맵

문제

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

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

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

 

https://www.acmicpc.net/problem/11478

11478번: 서로 다른 부분 문자열의 개수

 

import sys

S=list(sys.stdin.readline().rstrip())
sub=set() 

for x in range(len(S)):
  for y in range(x,len(S)):
    sub_list=S[x:y+1] 
    sub.add("".join(sub_list)) 

print(len(sub))

 

※ 집합과 맵

 

 

● set() 함수를 이용하여 sub라는 집합을 정의한다. (중복되는 요소는 자동으로 추가하지 않는다.)

 

● 그리고 가능한 모든 부분 문자열을 매 단계에서 sub_list에 저장한 뒤, join() 함수를 이용하여 sub_list안 부분 문자열의 요소들을 문자열로 합치고 sub에 add 해준다.

 

● 그 후 sub에 들어있는 길이를 출력해주면 서로 다른 부분 문자열의 개수가 출력된다!

 

 

결과