iterator
문제 설명
머쓱이는 태어난 지 6개월 된 조카를 돌보고 있습니다. 조카는 아직 "aya", "ye", "woo", "ma" 네 가지 발음을 최대 한 번씩 사용해 조합한(이어 붙인) 발음밖에 하지 못합니다. 문자열 배열 babbling이 매개변수로 주어질 때, 머쓱이의 조카가 발음할 수 있는 단어의 개수를 return하도록 solution 함수를 완성해주세요.
제한사항
1 ≤ babbling의 길이 ≤ 100
1 ≤ babbling[i]의 길이 ≤ 15
babbling의 각 문자열에서 "aya", "ye", "woo", "ma"는 각각 최대 한 번씩만 등장합니다.
즉, 각 문자열의 가능한 모든 부분 문자열 중에서 "aya", "ye", "woo", "ma"가 한 번씩만 등장합니다.
문자열은 알파벳 소문자로만 이루어져 있습니다.
내 풀이)
문제와 조건을 읽고, 순열로 풀어야 겠다고 판단했다. 그런데 itertools를 사용할 생각을 못해서 풀이가 막혀버렸고, 검색을 해서 itertools의 permutations을 알게 되어 활용해 문제를 풀었다.
# itertools에서 permutations를 import
from itertools import permutations
# 내가 만들고자하는 함수를 정의 ;def 함수이름(매개변수이름):
def solution(babbling): #babbling은 문자열 배열
answer = 0 # 단어의 개수
word = ["aya", "ye", "woo", "ma"]
p_words = [] # 조합된 단어를 담을 곳
for i in range(1,5):
for j in permutations(word,i):
p_words.append("".join(j))# 구분자.join(리스트)
for i in babbling:
if i in p_words:
answer += 1
return answer