백준 1759번 암호 만들기 C++

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

주어진 알파벳들을 오름차순으로 정렬하고 백트래킹을 이용하여 해결할 수 있는 문제이다.

오름차순으로 정렬된 알파벳들을 중복되지 않게 순서대로 탐색하였으며, 종료 조건은 정답을 저장하는 배열의 크기가 암호의 크기와 같을 때 종료하도록 설계하였다. 

한 가지 주의해야할 점은 암호는 한 가지의 모음과 두 가지의 자음이 필수적으로 있어야 한다. 따라서 정답을 출력할 때 그것을 확인하고 출력을 해야 한다.

 

#include <iostream>
#include <string>
#include <vector>
#include <math.h>
#include <algorithm>
#include <utility>
#include <stack>
#include <queue>
#include <math.h>
#include <set>
#include <unordered_set>

using namespace std;
typedef long long ll;

int l, c;

void BT(vector<char> v, vector<char> ans, int start) {
	if (ans.size() >= l) {
		bool isAns = false;
		int vowelCnt = 0;
		int consonantCnt = 0;
		for (auto a : ans) {
			if (a == 'a' || a == 'e' || a == 'i' || a == 'o' || a == 'u') {
				vowelCnt++;
			}
			else {
				consonantCnt++;
			}
			if (vowelCnt >= 1 && consonantCnt >= 2) {
				isAns = true;
				break;
			}
		}
		if (isAns) {
			for (auto a : ans){
				cout << a;
			}
			cout << "\n";
		}
		return;
	}
	for (int i = start; i < v.size(); i++) {
		ans.push_back(v[i]);
		BT(v, ans, i + 1);
		ans.pop_back();
	}
}

int main() {
	ios_base::sync_with_stdio(false);
	cin.tie(NULL);
	cout.tie(NULL);

	cin >> l >> c;
	vector<char> v(c);
	for (int i = 0; i < c; i++) {
		cin >> v[i];
	}
	sort(v.begin(), v.end());
	vector<char> ans;
	BT(v, ans, 0);
};