코딩테스트 입문 - 옹알이(1)

- 문제

 

- 배운 점

문자열 순열 생성

1. 순열 : nPr 순서O 중복X

from itertools import permutations

permutations(a,b) : aPb

 

2. 조합 : nCr 순서X 중복X

from itertools import combinations

combinations(a,b) : aCb

 

- 코드

from itertools import permutations

def solution(babbling):
    answer = 0
    bab = ["aya", "ye", "woo", "ma"]
    results=[]
    # pertmutations 사용해 순열 생성 (babPi)
    for i in range(1, len(bab)+1):
        for p in permutations(bab, i):
            results.append(''.join(p))     
            
    for j in babbling : 
        if j in results:
            answer+=1
    return answer

 

정규표현식 사용

- 첫문자가 aya, ye, woo, ma로 시작하고(^) 끝나는($) 단어

- +로 한번 이상 반복 

import re

def solution(babbling):
    regex = re.compile('^(aya|ye|woo|ma)+$')
    cnt=0
    for e in babbling:
        if regex.match(e):
            cnt+=1
    return cnt