백준 - 1759번(암호 만들기)

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

문제

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

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

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

import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.util.Arrays;
import java.util.StringTokenizer;

public class Main {

    private static String vowel = "aeiou";
    private static StringBuilder answer = new StringBuilder();

    public static void main(String[] args) throws IOException {
        BufferedReader reader = new BufferedReader(new InputStreamReader(System.in));
        StringTokenizer tokenizer = new StringTokenizer(reader.readLine());
        int L = Integer.parseInt(tokenizer.nextToken());
        int C = Integer.parseInt(tokenizer.nextToken());
        String[] target = reader.readLine().split(" ");

        Arrays.sort(target);

        combination(target, 0, 0, C, L, new StringBuilder(), 0, 0);

        System.out.println(answer);
    }

    // 대상 문자열 배열, 시작 인덱스, 현재 갯수, 배열 길이, 뽑아야하는 갯수, 문자열 저장용, 모음 수, 자음 수
    private static void combination(String[] target, int start, int currCnt, int totalCnt, int limit, StringBuilder sb, int vowelCnt, int consonantCnt) {
        if (currCnt == limit) {
            if (vowelCnt >= 1 && consonantCnt >= 2) answer.append(sb).append("\n");
            return;
        }

        for (int i = start; i < totalCnt; i++) {
            int currSize = sb.length();
            String temp = target[i];
            sb.append(target[i]);
            if (vowel.contains(temp)) combination(target, i + 1, currCnt + 1, totalCnt, limit, sb, vowelCnt + 1, consonantCnt);
            else combination(target, i + 1, currCnt + 1, totalCnt, limit, sb, vowelCnt, consonantCnt + 1);
            sb.setLength(currSize);
        }
    }

}

• 간단히 조합을 사용하여 풀 수 있었다.
• 사전순으로 나열하기 위해 배열을 정렬한 후 메서드를 호출하였다.
• 정해진 갯수만큼 뽑은 조합이 완성되면, 그 안에 자음과 모음의 수를 카운팅하여 조건을 충족하는 경우 출력하도록 하였다.