[Java] 백준 2011 암호코드

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

난이도

골드 5

문제

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

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

입력

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

출력

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

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

코드

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

class Main {
    public static void main(String[] args) throws IOException {
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        String code = br.readLine();
        int N = code.length();
        int[] dp = new int[N + 1];

        dp[0] = 1;
        if(code.charAt(0) == '0'){
            System.out.println(0);
            return;
        }

        dp[1] = 1;
        for(int i = 2; i <= N; i++){
            int one = code.charAt(i - 1);
            int two = code.charAt(i - 2);

            if(one >= '1' && one <= '9'){
                dp[i] = (dp[i] + dp[i - 1]) % 1000000;
            }

            int twoDigit = (two - '0')*10 + (one - '0');
            if(twoDigit >= 10 && twoDigit <= 26){
                dp[i] = (dp[i] + dp[i - 2]) % 1000000;
            }
        }
        
        System.out.println(dp[N]);
    }
}

문자열 s가 주어졌을 때

• s[i-1](한 자리)가 1~9 사이일 경우: dp[i] += dp[i-1]

• s[i-2]s[i-1](두 자리 숫자)가 10~26일 경우: dp[i] += dp[i-2]

• '0'은 단독으로 해석할 수 없고, 반드시 앞에 '1' 또는 '2'가 와야 유효함 (즉, 10, 20만 유효)

• 두 자리 수가 1~26 사이면 한 글자로도 해석 가능