[파이썬] 백준 1759 암호 만들기

숫자가 작아서 조합으로 풀었다.

import sys
from itertools import combinations
 
 
L,C = list(map(int,sys.stdin.readline().rstrip().split(" ")))
 
k = list(sys.stdin.readline().rstrip().split(" "))
 
mo = ['a','e','i','o','u']
 
 
temp = list(combinations(k,L))
 
answer = [ ]
 
for i in range(0,len(temp)):
 
    a = temp[i]
 
    momo = 0
    jaja = 0
 
    for j in range(0,len(a)):
 
        if a[j] in mo:
            momo += 1
        else:
            jaja += 1
 
    if momo>=1 and jaja>=2:
 
        a = list(sorted(a))
 
        answer.append("".join(a))
 
answer = list(sorted(answer))
 
for i in answer:
    print(i)