[Python] 백준 #1759- 암호 만들기

문제

---

 

1759번: 암호 만들기

코드

---

My answer

from itertools import combinations
import sys
input = sys.stdin.readline

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

a = list(combinations(letter, L))
a = ["".join(sorted(i)) for i in a]

for i in sorted(a):
    cnt = 0
    for j in vowels:
        if (j in i): cnt += 1
    if (len(i)-cnt >= 2 and cnt >= 1):
        print(i)

 

 

풀이

---

나는 단순하게, 조합으로 만들 수 있는 모든 경우를 구한 후, 자음과 모음의 수를 세어 걸러줬다.

문제를 풀며 중요했던 점은 우선 itertools의 combinations를 알아야만 했고, 다음으로 sorted를 통해서 미리 배열을 정렬하여 조건에 맞도록 만들어줬고, 최종적으로 자음과 모음의 개수까지 조건에 만족할 때 출력해줬다.