문제
문자열 S가 주어졌을 때, S의 서로 다른 부분 문자열의 개수를 구하는 프로그램을 작성하시오.
부분 문자열은 S에서 연속된 일부분을 말하며, 길이가 1보다 크거나 같아야 한다.
예를 들어, ababc의 부분 문자열은 a, b, a, b, c, ab, ba, ab, bc, aba, bab, abc, abab, babc, ababc가 있고, 서로 다른것의 개수는 12개이다.
난이도
Silver 3
입력 조건
첫째 줄에 문자열 S가 주어진다. S는 알파벳 소문자로만 이루어져 있고, 길이는 1,000 이하이다.
출력 조건
첫째 줄에 S의 서로 다른 부분 문자열의 개수를 출력한다.
입력 예시
ababc
출력 예시
12
처음 짠 코드
s = input()
s_list = []
for i in range(1, len(s)+1):
for j in range(len(s)-i+1):
s_list.append((s[j:j+i]))
print(len(set(s_list)))
수정 코드
s = input()
s_set = set()
for i in range(1, len(s)+1):
for j in range(len(s)-i+1):
s_set.add(s[j:j+i])
print(len(s_set))
느낀 점
처음에 그냥 막연히 리스트로 선언해놓고 나중에 set으로 만드는 방식으로 짰었는데
아예 처음부터 set()으로 선언해놓고 그대로 짜니까 시간을 1/2 이상 줄일 수 있었다.
더 효율적으로 짤 수 있는 방안을 고민하자.
리스트가 익숙하다고 리스트로 선언부터 하는 습관을 버리자 !!