머쓱이는 태어난 지 6개월 된 조카를 돌보고 있습니다. 조카는 아직 "aya", "ye", "woo", "ma" 네 가지 발음을 최대 한 번씩 사용해 조합한(이어붙인) 발음밖에 하지 못합니다. 문자열 배열 babbling이 매개변수로 주어질 때, 머쓱이의 조카가 발음할 수 있는 단어의 개수를 return하도록 solution 함수를 완성해 주세요
babbling의 길이 <= 100babbling[i]의 길이 <= 15babbling의 각 문자열에서 "aya", "ye", "woo", "ma"가 한 번씩만 등장합니다.| babbling | result |
|---|---|
| ["aya", "yee", "u", "maa", "wyeoo"] | 1 |
| ["ayaye", "uuuma", "ye", "yemawoo", "ayaa"] | 3 |
입출력 예 # 1
입출력 예 #2
babbling의 길이를 n, 각 문자열의 최대 길이를 l이라고 할 때, 전체 시간복잡도는 O(n × l)
(방법 3 : O(k)(k < l, k는 실제로 검사하게 되는 평균적인 문자 수))
방법 1: 문자열 치환(replace) + 중복 방지
def solution(babbling):
valid = ["aya", "ye", "woo", "ma"]
count = 0
for word in babbling:
temp = word
for v in valid:
temp = temp.replace(v, " ")
# 중복 방지를 위해 한 단어가 두 번 연속 사용된 경우 제거
if all(word.count(v*2) == 0 for v in valid) and temp.strip() == "":
count += 1
return count
방법 2: 정규 표현식 (re 모듈) + 중복 방지
import re
def solution(babbling):
valid = ["aya", "ye", "woo", "ma"]
count = 0
for word in babbling:
# 정규식으로 전체가 valid 단어들로만 구성되어 있는지 확인
if re.fullmatch(r"(aya|ye|woo|ma)+", word):
# 중복 단어 사용 방지
is_duplicate = any(v * 2 in word for v in valid)
if not is_duplicate:
count += 1
return count
방법 3: 시간복잡도 개선 (replace 방식 기반)
def solution(babbling):
valid = ["aya", "ye", "woo", "ma"]
count = 0
for word in babbling:
# 1. 길이 초과 시 즉시 탈락
if len(word) > 10:
continue
# 2. 중복된 단어가 연속된 경우 탈락
if any(v*2 in word for v in valid):
continue
# 3. 유효 단어만 replace하고 나머지 문자가 남으면 탈락
temp = word
for v in valid:
temp = temp.replace(v, " ")
if temp.strip() == "":
count += 1
return count