백준 1759 암호 만들기 with Python

www.acmicpc.net/problem/1759

1759번: 암호 만들기

---

<My code>

from itertools import combinations

l, c = map(int, input().split())
array = input().split()
array.sort()
case = combinations(array, l)
aeiou = ['a', 'e', 'i', 'o', 'u']

count = 0
for i in case:
    for j in range(5):
        if aeiou[j] in i:
            count += 1
    if count == 0:
        count = 0
        continue
    if count >= l-1:
        count = 0
        continue
    print(''.join(i))
    count = 0

combinations를 이용해서 풀었다.

정답을 받긴 했지만 코드가 너무 지저분해서 아쉬움이 남는다. 

 

 

<Reference code>

from itertools import combinations

vowels = ['a', 'e', 'i', 'o', 'u']
l, c = map(int, input().split())

array = input().split()
array.sort()

for password in combinations(array, l):
    count = 0
    for i in password:
        if i in vowels:
            count += 1
    if count >= 1 and count <= l - 2:
        print(''.join(password))

1. 조합을 변수에 삽입하는 과정 생략하고 바로 for 문에 삽입

2. count 변수 for 문 안에서 생성

조금 더 간결한 코드를 만들 수 있었다.