[백준] 1759번(암호 만들기) with Python

문제

www.acmicpc.net/problem/1759

1759번: 암호 만들기

풀이

배열을 정렬한 후 itertools 라이브러리에서 combinations 이터레이터를 사용해서 문제를 풀었다.

그리고 최소 한 개의 모음(a, e, i, o, u)과 최소 두 개의 자음으로 구성이 되었는지 카운트 해서 결과를 도출했다.

docs.python.org/ko/3/library/itertools.html#itertools.combinations

itertools — 효율적인 루핑을 위한 이터레이터를 만드는 함수 — Python 3.9.0 문서

<사용법 참고>

답

import sys 
from itertools import combinations

vowels = ['a', 'e', 'i', 'o', 'u']
L, C = map(int,input().split(' '))
pwd = sorted(input().split(' ')[:C]) 

comb = combinations(pwd, L)

for c in comb:
    cnt = 0
    for r in c:
        if r in vowels:
            cnt += 1

    if (cnt >= 1) and (cnt <= L-2):
        print(''.join(c))