[백준] 1759번 C/C++ 풀이 _ 암호 만들기



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

시간 제한메모리 제한제출정답맞은 사람정답 비율
2 초128 MB96144197298943.763%

문제

바로 어제 최백준 조교가 방 열쇠를 주머니에 넣은 채 깜빡하고 서울로 가 버리는 황당한 상황에 직면한 조교들은, 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

출처

  • 문제를 만든 사람: author5
  • 잘못된 데이터를 찾은 사람: djm03178
  • 데이터를 추가한 사람: sksdong1

알고리즘 분류

풀이

dfs 를 이용하여 문제를 해결합니다. 
set에 모음을 저장해 놓고 dfs를 진행할 때마다 모음인지 확인합니다. 

모음이면 모음의 수를 1개 늘려서 다음 dfs로 진행하고, 자음이면 자음의 수를 1개 늘려서 다음 dfs로 진행합니다. 
미리 들어온 알파벳을 정렬해놓고 dfs 를 진행하면 순서대로 출력됩니다. 
다음 dfs는 정렬된 알파벳에서 방금 탐색한 알파벳의 다음 알파벳 부터 탐색할 수 있게 인덱스의 정보를 함께 넘겨줍니다. 
depth가 n-1이 되면 모음이 1개 이상이고 자음이 2개 이상인지 확인해주고, 조건에 맞으면 출력해줍니다. 

자세한 풀이는 소스코드를 참조해주세요~

소스코드

#include <iostream>
#include <algorithm>
#include <vector>
#include <set>
using namespace std;
#define MOEUM 5
#define MAX_CHAR 16
// global
char char_arr[MAX_CHAR];
set<char> moeum_set({ 'a', 'e', 'i', 'o', 'u' });
vector<char> dfs_vec;
int L, C;

// 문자를 만들기 위한 dfs
void dfs(int _depth, int _moeum_num, int _jaeum_num, int _start_idx) {
    // dfs 조건 맞으면 문자 출력하기
    if (_depth == L - 1) {
        // 모음은 1개 이상 , 자음은 2개이상 확인
        if (_moeum_num < 1 || _jaeum_num < 2) return;
        // 출력
        for (int L_idx = 0; L_idx < L; L_idx++)
            cout << dfs_vec[L_idx];
        cout << "\n";
        return;
    }
    // 필수 조건 체크
    // 1. _start_idx 의 범위 초과시 종료
    // 2. 필요한 숫자의 개수가 남은 숫자의 개수보다 크면 끝내기
    if (_start_idx >= C) return;
    if (L - (_moeum_num + _jaeum_num) > C - _start_idx) return;

    for (int c_idx = _start_idx; c_idx < C; c_idx++) {
        dfs_vec.push_back(char_arr[c_idx]);
        // 자음이면
        if (moeum_set.find(char_arr[c_idx]) == moeum_set.end())
            dfs(_depth + 1, _moeum_num, _jaeum_num + 1, c_idx + 1);
        // 모음이면
        else
            dfs(_depth + 1, _moeum_num + 1, _jaeum_num, c_idx + 1);
        dfs_vec.pop_back();
    }
}

// main
int main() {
    ios::sync_with_stdio(false);
    cin >> L >> C;
    // input
    for (int c_idx = 0; c_idx < C; c_idx++) cin >> char_arr[c_idx];
    sort(char_arr, char_arr + C);

    for (int c_idx = 0; c_idx < C; c_idx++){
        dfs_vec.push_back(char_arr[c_idx]);
        // 자음이면
        if (moeum_set.find(char_arr[c_idx]) == moeum_set.end()) 
            dfs(0, 0, 1, c_idx + 1);
        // 모음이면
        else
            dfs(0, 1, 0, c_idx + 1);
        dfs_vec.pop_back();
    }

    return 0;
}