[코딩테스트] 백준 암호 만들기 파이썬(Python)

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

1759번: 암호 만들기

from itertools import combinations

alphabet = set([chr(i) for i in range(ord('a'), ord('z')+1)])
vowels = ['a', 'e', 'i', 'o', 'u']
consonants = list(alphabet - set(vowels))

l, c = map(int, input().split())
passwords = sorted(list(map(str, input().split())))
results = []
for password in combinations(passwords, l):
    vowel_count = 0
    cons_count = 0
    for c in password:
      if c in vowels:
        vowel_count += 1
        continue
      if c in consonants:
        cons_count += 1
    if vowel_count >= 1 and cons_count >= 2:
      results.append(''.join(password))

for result in results:
  print(result)

먼저 입력을 받기 전에 2개의 변수를 초기화하였습니다.

• vowels : 리스트, 모음
• consonats : 리스트, 자음

주어진 입력을 받은 후에는 sorted()함수를 이용하여 입력받은 문자를 정렬하였습니다.

그 다음은 itertools의 combinations 모듈을 이용하여 l개의 소문자로 이루어진 가능성이 있는 암호들의 모든 조합을 구합니다.

가능성 있는 암호들을 순회하면서 password를 이루는 알파벳이 모음이라면 vowel_count를 +1을 하고, 자음이라면 cons_count를 +1 하여

password가 몇개의 모음, 자음으로 구성되어 있는지 확인합니다.

password의 모든 알파벳을 확인하고 password가 아래의 조건에 부합하면 results에 추가합니다.

• vowel_count >= 1 : 최소 한 개의 모음으로 구성되어 있는지
• cons_count >= 2 : 최소 두 개의 자음으로 구성되어 있는지

results리스트의 원소들을 하나씩 출력하여 문제를 해결합니다.