[BOJ] 2011 - 암호코드 JAVA

www.acmicpc.net/problem/2011

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

 

 

 

 

 

풀이 .

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 pw = br.readLine();
        int n = pw.length();
        int[] dp = new int[n + 1];
        char[] ch = new char[n + 1];
        for(int i = 1; i <= n; i++) {
            ch[i] = pw.charAt(i-1);
        }

        dp[0] = 1;
        for(int i = 1; i <= n; i++) {
            // [i]가 알파벳 하나인 경우
            int num = ch[i] - '0';
            if(1 <= num && num <= 9) {
                dp[i] = dp[i-1];
                dp[i] %= 1000000;
            }
            if(i == 1) continue;

            // [i-1]+[i]가 알파벳 하나인 경우. 10~26
            num = (ch[i-1] - '0') * 10 + (ch[i] - '0');
            if(10 <= num && num <= 26) {
                dp[i] += dp[i-2];
                dp[i] %= 1000000;
            }
        }
        System.out.println(dp[n]);
    }
}

 

dp[n] = "n번째 숫자까지 검사했을 때 나오는 경우의 수"

 

1. n번째 숫자가 알파벳 하나인 경우 -> n번째 숫자가 0이 아니어야 함 (1~26 이니까)

2. n-1, n번째 숫자가 알파벳 하나인 경우 -> n-1, n번째 숫자를 이어붙인 것이 10~26 안에 들어와야 함

 

각 조건이 충족될 때 각각 dp[n-1], dp[n-2] 을 더해준다.

 

 

 

범위 설정에 주의하자

int x = ch[i] - '0';
if(1 <= x && x <= 9)
와
if('1' <= ch[i] && ch[i] <= '9')
는 같다.
.
.
int x = (ch[i-1] - '0') * 10 + (ch[i] - '0');
if(10 <= x && x <= 26)
와
if('1' <= ch[i-1] && ch[i-1] <= '2'  // 십의자리
        && '0' <= ch[i] && ch[i] <= '6') {  // 일의자리
는 다르다.

 

char를 바로 사용하려다가 위와 같은 실수를 범했다.

 

한자리 숫자를 검사하는 경우에서는 괜찮았지만 두자리 숫자를 검사할 때 문제가 생겼다.

 

저 경우에선 17, 18, 19 를 고려하지 못한다.