백준 1759 - 암호 만들기 (파이썬)

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

• 문제암호는 서로 다른 L개의 알파벳 소문자들로 구성되며 최소 한 개의 모음(a, e, i, o, u)과 최소 두 개의 자음으로 구성되어 있다고 알려져 있다. 또한 정렬된 문자열을 선호하는 조교들의 성향으로 미루어 보아 암호를 이루는 알파벳이 암호에서 증가하는 순서로 배열되었을 것이라고 추측된다. 즉, abc는 가능성이 있는 암호이지만 bac는 그렇지 않다.
• 새 보안 시스템에서 조교들이 암호로 사용했을 법한 문자의 종류는 C가지가 있다고 한다. 이 알파벳을 입수한 민식, 영식 형제는 조교들의 방에 침투하기 위해 암호를 추측해 보려고 한다. C개의 문자들이 모두 주어졌을 때, 가능성 있는 암호들을 모두 구하는 프로그램을 작성하시오.
• 바로 어제 최백준 조교가 방 열쇠를 주머니에 넣은 채 깜빡하고 서울로 가 버리는 황당한 상황에 직면한 조교들은, 702호에 새로운 보안 시스템을 설치하기로 하였다. 이 보안 시스템은 열쇠가 아닌 암호로 동작하게 되어 있는 시스템이다.
• 입력
• 첫째 줄에 두 정수 L, C가 주어진다. (3 ≤ L ≤ C ≤ 15) 다음 줄에는 C개의 문자들이 공백으로 구분되어 주어진다. 주어지는 문자들은 알파벳 소문자이며, 중복되는 것은 없다.
• 출력
• 각 줄에 하나씩, 사전식으로 가능성 있는 암호를 모두 출력한다.
• 예제
  • 입력
  • 4 6 a t c i s w
  • 출력
  • acis acit aciw acst acsw actw aist aisw aitw astw cist cisw citw istw

---

풀이

문제 상에서는 암호가 서로 다른 L개의 알파멧 소문자들로 구성되어 있으며, 최소 한개의 모음 및 두개의 자음으로 구성되어 있다고 했다.

또한 정렬된 문자열을 선호하는 조교들의 성향으로 미루어 보아 알파벳이 암호에서 증가하는 순서로 배열되어 있다고 추측했다는 말에서 힌트를 얻었다.

해당 문제는 먼저 정렬을 진행해야한다고 생각할 수 있는 부분이다.

지문과 같이 abc는 가능성이 있지만, bac는 그렇지 않다고 한것 처럼, C 가지의 문자의 종류를 바탕으로 가능성 있는 암호를 모두 구하는 프로그램을 작성하는 것이 해당 풀이의 목표이다.

그렇다면 C가지의 문자를 추출하는 방법은 itertools의 combinations 모듈을 사용하여 조합을 구하는 것이다.

전체적인 조합을 모두 구하면서(L개의 길이) 만큼 조합을 모두 진행한다.

예제에서 주어진 입력값을 정렬하면, a c i s t w 로 정렬된다. 여기서 4개를 조합하면서 진행하는 것이다.

acis acit aciw 등등등을 진행하면서, 모음이 1개 이상이며 자음이 2개이상인 친구들만 출력하면 되는 문제이다.

코드

# 백준 1759 - 암호 만들기
# 분류 : 브루트포스

import sys
from itertools import combinations

input = sys.stdin.readline

L, C = map(int, input().split())
A = input().split()

A.sort()

for combination in combinations(A, L) :
    moum = 0
    jaum = 0
    for a in combination :
        if a in {'a', 'e', 'i', 'o', 'u'} :
            moum += 1
        else :
            jaum += 1
    
    if moum >= 1 and jaum >= 2 :
        print("".join(combination))