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

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

1759번: 암호 만들기

 

처음에는 문자를 하나하나 비교해서 풀어야 하나 라는 생각을 했다.

귀찮기도 하고 감도 잘 안 잡혀서 생각 좀 하다가 java 풀이를 검색했다.

combination을 사용한 풀이가 있길래 나도 combination 함수를 사용하였다.

 

combination 함수 사용하기 전에 입력받은 알파벳들 정렬하는 과정을 빼먹어서 시간이 좀 걸렸다.

 

문자열 합칠 때 join 함수 사용하는 것도 잊지말자.

 

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

l, c = map(int, input().split())
char = list(map(str, input().split()))
char.sort()

comb = combinations(char, l)
vowel = ['a', 'e', 'i', 'o', 'u']
for i in comb:
    csn = 0 #consonant
    vw = 0 #vowel
    for j in range(l):
        if i[j] in vowel:
            vw += 1
        else:
            csn += 1
    if csn >= 2 and vw >= 1:
        print(''.join(i))