1759 암호 만들기

BOJ::1759 암호 만들기

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

백트래킹을 통해 결과에 쓰일 String만 만들어주면 된다.

백트래킹을 처음 접했을 때 연습하기 좋은 문제 인거 같다.

<JAVA>

import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.util.Arrays;
import java.util.StringTokenizer;
 
public class Main {
    
    static int l,c;
    static String a[];
    static int con,vow,cnt;
    static StringBuilder sb = new StringBuilder();
 
    public static void main(String args[]) throws NumberFormatException, IOException
    {
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        StringTokenizer st = new StringTokenizer(br.readLine());
        l = Integer.parseInt(st.nextToken());
        c = Integer.parseInt(st.nextToken());
        String s = br.readLine();
        a=s.split(" ");
        Arrays.sort(a);
        
        for(int i = 0 ; i < c-l+1;i++){
            cnt=0;
            con=0;
            vow=0;
            dfs(i,a[i]+"");
        }
        
        System.out.print(sb.toString());
    }
    
    public static void dfs(int v,String str){
        cnt++;
        
        if(isVow(a[v])){
            vow++;
        }else{
            con++;
        }
        if(cnt==l){
            if(con>=2 && vow>=1){
                sb.append(str + "\n");
            }
        }else{
            for(int i = v+1; i<c ; i++){
                dfs(i,str+a[i]+"");
            }
        }
        
        cnt--;
        if(isVow(a[v])){
            vow--;
        }else{
            con--;
        }
        
    }
    //모음 확인
    public static boolean isVow(String s){
        if(s.equals("a") || s.equals("e") || s.equals("i") || s.equals("o") || s.equals("u")){
            return true;
        }
        return false;
    }
}

C++

#include <iostream>
#include <string>
#include <algorithm>
using namespace std;
 
int l, c,con,vow,cnt;
string result;
string a[16];
 
bool isVow(string s) {
    if (s == "a" || s == "e" || s == "i" || s == "o" || s == "u") {
        return true;
    }
    return false;
}
 
void dfs(int v, string str) {
    cnt++;
    if (isVow(a[v])) {
        vow++;
    }
    else {
        con++;
    }
 
    if (cnt == l) {
        if (vow >= 1 && con >= 2) {
            result += str + "\n";
        }
    }
    else {
        for (int i = v + 1; i < c; i++) {
            dfs(i, str + a[i] + "");
        }
    }
 
    cnt--;
    if (isVow(a[v])) {
        vow--;
    }
    else {
        con--;
    }
}
 
int main() {
    result = "";
    cin >> l >> c;
    for (int i = 0; i < c; i++) {
        cin >> a[i];
    }
    sort(a,a+c);
 
 
    for (int i = 0; i < c - l + 1; i++) {
        cnt = 0;
        vow = 0;
        con = 0;
        dfs(i, a[i]);
    }
 
    cout << result;
 
}