[프로그래머스][파이썬] LV.1 옹알이 (2)

문제

머쓱이는 태어난 지 11개월 된 조카를 돌보고 있습니다. 조카는 아직 "aya", "ye", "woo", "ma" 네 가지 발음과 네 가지 발음을 조합해서 만들 수 있는 발음밖에 하지 못하고 연속해서 같은 발음을 하는 것을 어려워합니다. 문자열 배열 babbling이 매개변수로 주어질 때, 머쓱이의 조카가 발음할 수 있는 단어의 개수를 return하도록 solution 함수를 완성해주세요.

제한사항

• 1 ≤ babbling의 길이 ≤ 100
• 1 ≤ babbling[i]의 길이 ≤ 30
• 문자열은 알파벳 소문자로만 이루어져 있습니다.

내 풀이

def solution(babbling):
    answer = 0
    can_do = ["aya", "ye", "woo", "ma"]
    repeats = ["ayaaya", "yeye", "woowoo", "mama"]

    # 길이가 다르기 때문에 zip은 사용할 수 없음
    # babbling 문자열 배열 
    for i in babbling:
        for d in repeats:
            i = i.replace(d, 'X')
        for d in can_do:
            i = i.replace(d, 'O')
        
        flag = True
        
        for j in i:
            if j != 'O':
                flag = False
                break
                
        if flag == True:
            answer += 1
    
    return answer

• 같은 발음이 연속될 수 없으니 해당되는 발음을 list로 만들어서 repeats에 저장
• 문자열의 길이가 다르기 때문에 zip 함수는 사용할 수 없음 => 중첩 for문 사용
• 먼저 repeats를 돌리면서 안되는 발음을 X로 전환
• 가능한 발음은 O로 전환
• 그렇게 새로 만들어진 문자열을 for으로 돌면서 'O'가 아닐 경우 flag를 false로 전환하며 break
• flag가 True일 경우 answer에 +1

다른 풀이

def solution(babbling):
    answer = 0
    for i in babbling:
        for j in ['aya','ye','woo','ma']:
            if j*2 not in i:
                i=i.replace(j,' ')
        if len(i.strip())==0:
            answer +=1
    return answer

• j*2를 사용해 중복 확인
• strip을 사용해 인자로 전달된 문자를 문자열의 왼쪽과 오른쪽에서 제거.