[BOJ] 암호 만들기

문제

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

이 보안 시스템은 열쇠가 아닌 암호로 동작하게 되어 있는 시스템이다.

암호는 서로 다른 L개의 알파벳 소문자들로 구성되며 최소 한 개의 모음(a, e, i, o, u)과 최소 두 개의 자음으로 구성되어 있다고 알려져 있다.

또한 정렬된 문자열을 선호하는 조교들의 성향으로 미루어 보아 암호를 이루는 알파벳이 암호에서 증가하는 순서로 배열되었을 것이라고 추측된다.

즉, abc는 가능성이 있는 암호이지만 bac는 그렇지 않다.

새 보안 시스템에서 조교들이 암호로 사용했을 법한 문자의 종류는 C가지가 있다고 한다. 이 알파벳을 입수한 민식,

영식 형제는 조교들의 방에 침투하기 위해 암호를 추측해 보려고 한다. C개의 문자들이 모두 주어졌을 때, 가능성 있는 암호들을 모두 구하는 프로그램을 작성하시오.

풀이

암호를 만들기 위해 여러 조건을 만족해야한다.
우선 암호를 한번씩 사용해서 총 L 길이의 암호를 만족해야 하기 때문에 순열이나 조합을 이용해야 한다.
출력 힌트를 보면 조합을 이용해야 함을 알 수 있다.

조합을 구현하면서 지켜야 할 조건은 모음과 자음의 갯수 제한이다.
따라서 Depth체크를 하면서 return 조건을 체크 하면서 동시에 모음고 자음의 갯수 체크를 해준다.

• 갯수 체크는 Colletion의 retainAll을 이용하여 하였다.

물론 이 조건을 만족하기 전 사전순 정렬이 필요하다.
조건을 만족하는 답을 구한 후에 정렬시키는 것은 시간 초과를 유발할 수 있다.
주어진 힌트를 먼저 정렬시킨 후에 조합을 수행하는 것은 이 후의 정렬을 할 필요 없이 자동 정렬이 되므로 더 효율적인 방법이다.

import java.lang.*;
import java.io.*;
import java.util.*;


class Main {
    static int L;
    static int C;
    static String[] hints;
    static boolean[] visited;
    static List<String> vowel = Arrays.asList("a", "e", "i", "o", "u");
    static List<String> answer = new ArrayList<>();
    public static void main(String[] args) throws Exception {
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        String[] str = br.readLine().split(" ");
        L = Integer.parseInt(str[0]);
        C = Integer.parseInt(str[1]);
        hints = new String[C];
        visited = new boolean[C];
        String[] guessCode = new String[L];

        str = br.readLine().split(" ");
        for(int i = 0; i < C; i++) {
            hints[i] = str[i];
        }
        Arrays.sort(hints);
        combine(guessCode, 0, 0);

        StringBuilder sb = new StringBuilder();
        for(String s : answer) {
            sb.append(s + "\n");
        }

        System.out.println(sb);
    }

    static void combine(String[] guessCode, int depth, int start) {
        if (depth == L) {
            if (check(guessCode)) {
                answer.add(String.join("", guessCode));
            }
            return;
        }

        for(int i = start; i < C; i++) {
            if (!visited[i]) {
                guessCode[depth] = hints[i];
                visited[i] = true;
                combine(guessCode, depth + 1, i);
                visited[i] = false;
            }
        }
    }

    static boolean check(String[] guessCode) {
        List<String> list1 = new ArrayList<>(vowel);
        List<String> list2 = new ArrayList<>(Arrays.asList(guessCode));

        list1.retainAll(list2);
        return (list1.size() >= 1) && (L - 2 >=  list1.size());
    }
}

https://github.com/Win-9/Algorism/tree/main/BOJ/%EC%95%94%ED%98%B8%EB%A7%8C%EB%93%A4%EA%B8%B0