[백준/C++]1759번: 암호 만들기

문제

 

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

문제

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

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

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

입력

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

출력

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

---

Algorithm

먼저 알파벳 배열을 입력받는다 이후 암호는 알파뱃이 증가하는 순서 및 사전순으로 모든 경우의 수를 나열하라 하였으므로 알파뱃 배열을 정렬한다. 이후 백트레킹을 활용해서 자리마다 전의 자리가 찾은 인덱스의 뒤부터 찾아 정답 배열에 삽입하는 L개 중첩된 for문이 작성된 것처럼 배열에 삽입한다. 만약 삽입한 알파벳이 모음이라면 모음 카운트를+1 아닌경우 자음카운트를+1를 인자로 전달하고 L개모두 채웠다면 자음이 2개이상 모음이 1개이상 사용되었다면 출력하고 아닌경우 출력하지 않도록하여 정답이 아닌경우를 걸러내었다.

 

---

 

 

Code

#include<iostream>
#include<algorithm>
using namespace std;

char ans[15];

void recur(int n,int c,int start,char *arr,int cnt,int v_cnt,int c_cnt);
bool is_vowel(char c);

int main(){
    int l,c;
    cin>>l>>c;
    char * arr=new char[c];
    for(int i=0;i<c;i++){
        cin>>arr[i];
    }
    sort(arr,arr+c);
    recur(l,c,0,arr,0,0,0);
    delete[] arr;
    return 0;
}

void recur(int n,int c,int start,char *arr,int cnt,int v_cnt,int c_cnt){
    if(cnt==n){
        if(v_cnt>0&&c_cnt>=2){
            for(int i=0;i<n;i++){
                cout<<ans[i];
            }
            cout<<'\n';
        }
        return;
    }
    for(int i=start;i<c-n+cnt+1;i++){
        if(is_vowel(arr[i])){
            ans[cnt]=arr[i];
            recur(n,c,i+1,arr,cnt+1,v_cnt+1,c_cnt);
        }
        else{
            ans[cnt]=arr[i];
            recur(n,c,i+1,arr,cnt+1,v_cnt,c_cnt+1);
        }
    }
}
bool is_vowel(char c){
    return c=='a'||c=='e'||c=='i'||c=='o'||c=='u';
}