옹알이(1)

프로그래머스

머쓱이는 태어난 지 6개월 된 조카를 돌보고 있습니다.
조카는 아직 “aya”, “ye”, “woo”, “ma” 네 가지 발음을
최대 한 번씩 사용해 조합한(이어 붙인) 발음밖에 하지 못합니다.
문자열 배열 babbling이 매개변수로 주어질 때,
머쓱이의 조카가 발음할 수 있는 단어의 개수를 return하도록
solution 함수를 완성해주세요.

# input setting
problem_num = '옹알이(1)'

import os, sys
path = os.getcwd() + f'\\txt\\{problem_num}'  + '.txt'
sys.stdin = open(path,'r')
babbling = sys.stdin.readline().strip().split(',')

def solution(babbling):
    from itertools import permutations as per
    answer = 0
    possibles = ["aya", "ye", "woo", "ma"]
    for i in range(2,5):
        possibles.extend(''.join(x).strip() for x in per(["aya", "ye", "woo", "ma"],i))
    for i in babbling:
        if i.strip() in possibles:
            answer += 1
    return answer

print(solution(babbling))

3

• itertools.permutations

• strip