코딩 테스트/백준

[백준1759] 암호 만들기

땅어 2020. 3. 9. 16:36
728x90
300x250

문제

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

 

1759번: 암호 만들기

첫째 줄에 두 정수 L, C가 주어진다. (3 ≤ L ≤ C ≤ 15) 다음 줄에는 C개의 문자들이 공백으로 구분되어 주어진다. 주어지는 문자들은 알파벳 소문자이며, 중복되는 것은 없다.

www.acmicpc.net

 

생각의 흐름

  1. 구해야 하는 것이 조합임을 파악
  2. 사전순으로 출력하기 위해 정렬
  3. 최소 한 개의 모음과 두 개의 자음을 포함하는지 조건 검사해서 출력

 

코드

from itertools import combinations

L, C = map(int, input().split())
alphabet = list(map(str, input().split()))
alphabet.sort()
moeum_list = ['a', 'e', 'i', 'o', 'u']
for case in combinations(alphabet, L):
    moeum = 0
    jaeum = 0
    for i in range(len(case)):
        if case[i] in moeum_list:
            moeum += 1
        else:
            jaeum += 1
    if moeum>0 and jaeum>1:
        print(''.join(map(str, case)))

 

728x90
300x250