[구름톤 챌린지] 6일차 - 문제 6. 문자열 나누기 (Python)

구름LEVEL

---

문제 6. 문자열 나누기

해당 문제는 문자열을 분리하는 모든 경우의 수를 조합으로 탐색한 후, 조건에 따라서 점수를 측정하는 완전 탐색 문제입니다.

 

 

import itertools
N = int(input()) # 문자열 길이
S = input() # 문자열

sub = []

for i in range(1, N-1):
	for j in range(i+1, N):
		sub.append([S[:i], S[i:j], S[j:]]) # 3개의 부분문자열을 담은 이중 리스트
		
P = sorted(list(set(list(itertools.chain(*sub))))) # 이중 리스트를 flatten 한 후 중복 제거

scores = [] # 부분문자열 별 스코어를 담을 리스트

for s in sub:
	score = 0
	score += P.index(s[0]) + 1
	score += P.index(s[1]) + 1
	score += P.index(s[2]) + 1
	scores.append(score)
	
print(max(scores)) # 최대 점수 출력

• 뽑을 수 있는 3개의 부분문자열을 이중 반복문을 통해서 모두 구한 후 이중 리스트 형태로 담았다.
• 문제 조건에 따라 중복을 제거하기 위해 이중 리스트를 flatten 시킨 후 사전순으로 정렬하였다. (itertools.chain() 사용)
• 정렬 결과에 따라 만들었던 부분문자열의 스코어들을 계산하여 전체 스코어 리스트를 만들고, 그중 가장 큰 최대 점수를 출력했다.