BOJ 2011 암호코드

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

문제

상근이와 선영이가 다른 사람들이 남매간의 대화를 듣는 것을 방지하기 위해서 대화를 서로 암호화 하기로 했다. 그래서 다음과 같은 대화를 했다.

• 상근: 그냥 간단히 암호화 하자. A를 1이라고 하고, B는 2로, 그리고 Z는 26으로 하는거야.
• 선영: 그럼 안돼. 만약, “BEAN”을 암호화하면 25114가 나오는데, 이걸 다시 글자로 바꾸는 방법은 여러 가지가 있어.
• 상근: 그렇네. 25114를 다시 영어로 바꾸면, “BEAAD”, “YAAD”, “YAN”, “YKD”, “BEKD”, “BEAN” 총 6가지가 나오는데, BEAN이 맞는 단어라는건 쉽게 알수 있잖아?
• 선영: 예가 적절하지 않았네 ㅠㅠ 만약 내가 500자리 글자를 암호화 했다고 해봐. 그 때는 나올 수 있는 해석이 정말 많은데, 그걸 언제 다해봐?
• 상근: 얼마나 많은데?
• 선영: 구해보자!

어떤 암호가 주어졌을 때, 그 암호의 해석이 몇 가지가 나올 수 있는지 구하는 프로그램을 작성하시오.

---

입력

첫째 줄에 5000자리 이하의 암호가 주어진다. 암호는 숫자로 이루어져 있다.

---

출력

나올 수 있는 해석의 가짓수를 구하시오. 정답이 매우 클 수 있으므로, 1000000으로 나눈 나머지를 출력한다.

암호가 잘못되어 암호를 해석할 수 없는 경우에는 0을 출력한다.

---

예제 입력 1

25114

예제 출력 1

6

예제 입력 2

1111111111

예제 출력 2

89

---

코드

import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;

public class Main {

    public static void main(String[] args) throws IOException {
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        String cipher = br.readLine();

        // 첫 숫자가 0일 경우, 해석 가능한 방법이 없으므로 0출력
        if(cipher.charAt(0) == '0') {
            System.out.println(0);
            return;
        }

        /**
         * 해석한 마지막 숫자가 한 개인 해석 방법 수를 저장한 배열은 one
         * 마지막 숫자가 두 개인 해석 방법 수를 저장한 배열은 two
         * ex) one -> (5, 12, 6), (4, 8), ...
         *     two -> (2, 21, 12), (4, 16), ...
         */
        int[] one = new int[cipher.length()], two = new int[cipher.length()];
        // 숫자가 1개일 경우, 해석 방법 수는 1가지
        one[0] = 1;

        for (int i = 1; i < cipher.length(); i++) {
            // 검사하고 있는 숫자가 0일 경우,
            if(cipher.charAt(i) == '0') {
                char last = cipher.charAt(i - 1);
                // 0은 단독으로 해석할 수 없기 때문에, 이전 문자가 1이나 2와 같이 0과 묶여 해석 가능할 경우
                // 이번 단계의 two의 해석 방법 수를 이전 단계 one의 해석 방법 수로 설정
                if(last == '1' || last == '2') two[i] = one[i - 1];
                    // 이전 문자가 1이나 2가 아니면 해석이 불가능하기 때문에 0출력
                else {
                    System.out.println(0);
                    return;
                }
                continue;
            }

            // 이번 단계의 one은 이전 단계의 one, two에 검사하고 있는 숫자 한 개만 붙이면 되기 때문에 이전 단계 one + two로 설정
            one[i] = one[i - 1] + two[i - 1];
            if(one[i] > 1_000_000) one[i] %= 1_000_000;
            // 이전 숫자와 같이 묶어 해석 가능할 경우, 이번 단계 two를 이전 단계 one으로 설정
            if(Integer.parseInt(cipher.substring(i - 1, i + 1)) < 27) two[i] = one[i - 1];
        }

        int answer = one[cipher.length() - 1] + two[cipher.length() - 1];
        System.out.println(answer % 1_000_000);
    }
}