Algorithm PS/BOJ
[BOJ] 1759 / 암호 만들기 / Python 파이썬
migrationArc
2021. 8. 4. 22:13
[BOJ] 1759 / 암호 만들기 / Python 파이썬
https://www.acmicpc.net/problem/1759
1759번: 암호 만들기
첫째 줄에 두 정수 L, C가 주어진다. (3 ≤ L ≤ C ≤ 15) 다음 줄에는 C개의 문자들이 공백으로 구분되어 주어진다. 주어지는 문자들은 알파벳 소문자이며, 중복되는 것은 없다.
www.acmicpc.net
풀이
경우의 수를 조합으로 구성하여 쉽게 해결하였다.
from itertools import combinations
L, C = map(int, input().split())
alphas = input().split()
alphas.sort()
aeiou = ('a', 'e', 'i', 'o', 'u')
# 최소 한 개의 모음(a, e, i, o, u)과 최소 두 개의 자음
# 알파벳이 암호에서 증가하는 순서로 배열
res = set()
for com in combinations(range(C), L):
tmp = ""
aeiouCnt = 0
notAeiouCnt = 0
for c in com:
if alphas[c] in aeiou:
aeiouCnt += 1
else:
notAeiouCnt += 1
tmp += alphas[c]
if aeiouCnt < 1 or notAeiouCnt < 2:
continue
res.add(tmp)
res = list(res)
res.sort()
for r in res:
print(r)