[Python] 코딩테스트 입문 : 옹알이 (1)

 

문제

---

코드

# "aya", "ye", "woo", "ma" 
# 네 가지 발음을 최대 한 번씩 사용해 조합한(이어 붙인) 발음밖에 하지 못합니다
# 문자열 배열 babbling이 매개변수로 주어질 때, 
# 머쓱이의 조카가 발음할 수 있는 단어의 개수를 return
def solution(babbling):
    answer = 0
    
    word_ok = ["aya", "ye", "woo", "ma"] 
    for e in babbling:
        while True:
          if e[:3] in word_ok:
              e = e[3:]
          elif e[:2] in word_ok:
              e = e[2:]
          else:
              if len(e) == 0:
                answer += 1
              
              break
    
    return answer

 

 

 

---