[백준/1759] 암호 만들기

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

1759번: 암호 만들기

---

 암호의 순서는 언제나 오름차순 정렬로 고정되어 있으므로, 순서에 대해 고려하지 않아도 됩니다. 따라서 조합(Combination)을 이용하여 간단하게 문제를 풀 수 있습니다. 파이썬 내장 라이브러리 중 하나인 itertools의 combinations 함수를 사용하면 됩니다.

 소스코드는 다음과 같습니다.

from itertools import combinations

l, c = list(map(int, input().split()))
array = input().split()

vowel = ('a', 'i', 'u', 'e', 'o')

array = list(combinations(array, l))
result = []
for i in array:
    v = 0
    for j in i:
        if j in vowel:
            v += 1
    if v >= 1 and l - v >= 2:
        result.append(''.join(sorted(i)))

for i in sorted(result):
    print(i)

 모음이 최소 1개, 자음이 최소 2개 있어야 한다는 제약이 있으므로 먼저 모음을 튜플로 초기화하여 관리하겠습니다. 이후 입력으로 받은 문자들에 대해서 길이 4개짜리 조합을 모두 구합니다. 조합 경우의 수를 순회하며 모음의 개수를 세고, 만약 모음의 개수가 1 이상이고, 자음의 개수(전체 길이에서 모음의 개수를 뺀 것)가 2 이상이라면 result 변수에 삽입합니다. 

 마지막으로 result변수를 오름차순 정렬한 뒤 순회하며 출력하면, 정답을 구할 수 있습니다.