프로그래머스 [lv2] 모음사전 파이썬

📌  접근

1. A, E, I, O, U 의 중복을 허용하며 모든 경우의 순열(permutation)을 만들어 낸다.

2. 각 모음의 조합이 1가지인 경우, 2가지인 경우, ... 5가지인 경우, 이렇게 각각의 경우에 맞는 모든 조합을 만들어야한다.

   2 - 1. 예를 들어 조합이 5가지인 경우는, AEIOU 뿐만이 아니라 (중복 허용하기 때문에) AAAAA, EEEEE ... UUUUU 같은 경우도 존재한다.

   2 - 2. 때문에 단순 permutation 메서드를 사용하면 안된다. 

3. 오름차순 정렬한다(소팅).

 

결론:  사실상, A, E, I, O, U 를 조합하여, 모든 경우를 쫙 펼친 후 소팅.

❌  permutation 으로 접근했을 때 (오답)

from itertools import product, chain


def solution(word):
    basic = ["A", "E", 'I', "O", "U"]
    word_set = basic[:]

    for i in range(2, 6):
        for j in permutations(basic, i):
            word_set.append(''.join(j))
   
	...

이 과정에서 permutation 과 product의 차이를 느꼈다.

permutation 은 각 원소를 한번씩 조합하는 반면  <=>  product 는 각 원소를 여러번 조합한다.

즉 product 를 사용하면 하나의 모음을 중복하여 조합할 수 있는 반면, permutation 은 하나의 모음에 대한 중복 연산은 처리하지 않는다.

때문에 product 를 사용하는 것이 옳다.

 

 

✅   itertools.product() 사용

from itertools import product, chain


def solution(word):
    basic = ["A", "E", 'I', "O", "U"]
    word_set = basic[:]
    for i in range(2, 6):
        word_set += list(map(list, product(basic, repeat=i)))
    result = sorted(list(map(lambda x: ''.join(x), word_set)))
    
    return result.index(word) + 1

 

+ itertools.chain() 메서드 사용

itertools 의 chain() 메서드를 사용하면 iterable한 객체들을 인수로 받아 하나의 iterator로 반환한다.

from itertools import product, chain


def solution(word):
    basic = ["A", "E", 'I', "O", "U"]
    word_set = basic[:]
    
    for i in range(2, 6):
        word_set = chain(word_set, list(map(list, product(basic, repeat=i))))
    result = sorted(list(map(lambda x: ''.join(x), word_set)))
    
    return result.index(word) + 1