[백준 1759] 암호 만들기

 주어진 알파벳을 정렬하고 (사전 순으로 출력하기 위해서) 갯수만큼 뽑아낸 이후에,

 모음이 1개 이상 && 자음이 2개 이상인지  검사하여

 정답이 가능한 모든 경우를 출력하는 코드를 작성하는 문제이다.

 솔직히 분류 상으로는 "백트레킹"이지만, 그렇게 하지 않아도 충분히 풀린다.

 기회가 되면 재귀함수로 구현할 예정.

 개인적인 생각으로는 방법은 두가지가 떠올랐다.

 

 1. STL의 next_permutation을 사용한 조합

참고하면 좋은 글=> (next_permutation으로 조합 만들기?)

=> 모음 자음 검사가 귀찮고, 정렬을 해줘야 하기 때문에 코드가 복잡해지기 쉽다.

( 내가 사용한 방법이다 )

2. 재귀 함수

=> 재귀 함수와 변수들을 적절하게 사용하면, 정답 조건을 한번에 만족시킬 수 있다.

(추가) 3. DFS를 사용

=> 바로 떠오르지는 않았던 구현 방법이다.

a,b,c (주어진) 알파벳으로 만들 수 있는 경우의 수를

모두 만들고 정답 조건에 부합하는지 확인하는 법이다.

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

문제

바로 어제 최백준 조교가 방 열쇠를 주머니에 넣은 채 깜빡하고 서울로 가 버리는 황당한 상황에 직면한 조교들은, 702호에 새로운 보안 시스템을 설치하기로 하였다. 이 보안 시스템은 열쇠가 아닌 암호로 동작하게 되어 있는 시스템이다.

암호는 서로 다른 L개의 알파벳 소문자들로 구성되며 최소 한 개의 모음과 최소 두 개의 자음으로 구성되어 있다고 알려져 있다. 또한 정렬된 문자열을 선호하는 조교들의 성향으로 미루어 보아 암호를 이루는 알파벳이 암호에서 증가하는 순서로 배열되었을 것이라고 추측된다. 즉, abc는 가능성이 있는 암호이지만 bac는 그렇지 않다.

새 보안 시스템에서 조교들이 암호로 사용했을 법한 문자의 종류는 C가지가 있다고 한다. 이 알파벳을 입수한 민식, 영식 형제는 조교들의 방에 침투하기 위해 암호를 추측해 보려고 한다. C개의 문자들이 모두 주어졌을 때, 가능성 있는 암호들을 모두 구하는 프로그램을 작성하시오.

입력

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

출력

각 줄에 하나씩, 사전식으로 가능성 있는 암호를 모두 출력한다.

예제 입력 1

4 6
a t c i s w

예제 출력 1

acis
acit
aciw
acst
acsw
actw
aist
aisw
aitw
astw
cist
cisw
citw
istw

소스코드

#include<iostream>
#include<algorithm>
#include<vector>
#include<string>
 
using namespace std;
 
vector<char> list;            // 모든 알파벳이 들어가는 컨테이너 
vector<int> idx;            // 조합 순열을 사용하기 위한 컨테이너 
vector<string> ans;            // 모든 정답을 저장하기 위한 컨테이너 
 
// 모음이 1개 이상이고 자음이 2개 이상인지 검사 
bool pass(string str) {
    int len = str.length();
    int mo = 0, ja = 0;                         // 모음 자음 카운트
    for (int i = 0; i < len; i++) {
        if (str[i] == 'a' || str[i] == 'e' || str[i] == 'i' || str[i] == 'o' || str[i] == 'u')
            mo++;
        else
            ja++;
    }
    return (mo > 0 && ja > 1) ? true : false;
}
 
// int 오른차순 정렬 
bool asc(int a, int b) {
    return a < b;
}
// string 오름차순 정렬  
bool cmp(string a, string b) {
    return a < b;
}
int main(void) {
    int choice, n;
    cin >> choice >> n;
    
    for (int i = 0; i < n; i++) {
        char tmp;
        cin >> tmp;
        list.push_back(tmp);
    }
    
    // 0과 1을 활용한 조합 계산 
    for (int i = 0; i < choice; i++)
        idx.push_back(1);
    for (int i = 0; i < n - choice; i++)
        idx.push_back(0);
 
    sort(list.begin(), list.end(), asc);
    sort(idx.begin(), idx.end(), asc);
 
    // 조합을 구하여 choice 개수만큼 원소를 추출한다. 
    do {
        string str = "";
        for (int i = 0; i < n; i++) {
            if(idx[i] == 1)                // 조합 
                str += list[i]; 
        }
        if (pass(str))                    // 모음이 1개 이상, 자음이 2개 이상인 경우 
            ans.push_back(str);            // 정답에 push
    } while (next_permutation(idx.begin(), idx.end(), asc));        // 다음 조합 순열 연산 
    
    sort(ans.begin(), ans.end(), cmp);                                // 정답이 담긴 vector<stirng> 컨테이너를 오름차순으로 정렬한다. 
 
    for (unsigned int i = 0; i < ans.size(); i++)                            // 출력 
        cout << ans[i] << '\n';
        
    return 0;
}

※ 본 글은 개인 포트폴리오 혹은 공부용으로 사용하기 때문에, 무단 복사 유포는 금지하지만, 개인 공부 용도로는 얼마든지 사용하셔도 좋습니다.