[Python] 백준 1759번: 암호 만들기

출처 : https://www.acmicpc.net/problem/1759

1759번: 암호 만들기

 

 

 

---

Code

from itertools import combinations
 
L, C = map(int, input().split())
word_list = set(input().split())
 
## A(=모음), B(=자음)
A = set(['a', 'e', 'i', 'o', 'u'])
B = set(['b','c','d','f','g','h','j','k','l','m','n','p','q','r','s','t','v','w','x','y','z'])
 
## word_list의 모음
word_list_A = word_list & A
 
for next in combinations(sorted(word_list), L):
    num = len(set(next) & word_list_A)
    if num == 0 or L-num < 2: continue
    print(''.join(next))