[백준] 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자리 글자를 암호화 했다고 해봐. 그 때는 나올 수 있는 해석이 정말 많은데, 그걸 언제 다해봐?
• 상근: 얼마나 많은데?
• 선영: 구해보자!

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

 

입출력

25114

 

6

---

아이디어

 

 

DP[i] // i번째 수까지 고려했을 때 가능한 해석의 수 

dp[0] = 1 // 빈 문자열 해석 1개

dp[1] = 1 // 첫글자 1개 

 

코드 

 

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

public class Main {
	static int N;
	static String str;

	public static void main(String[] args) throws IOException {
		BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
		str = br.readLine();
		N = str.length();
		if (str.charAt(0) == '0') {
			System.out.println(0);
			return;
		}
		int[] dp = new int[N + 1];// i번째 수까지 고려했을 때 가능한 해석의 수
		dp[0] = 1; // 빈 문자열 해석 1개
		dp[1] = 1; // 첫글자 1개
		for (int i = 2; i <= N; i++) {
			// 한자리 또는 두자리
			char one = str.charAt(i - 1);
			int twoLen = Integer.parseInt(str.substring(i - 2, i));

			if (one != '0') {
				dp[i] = (dp[i] + dp[i - 1]) % 1000000;
			}
			if (twoLen >= 10 && twoLen <= 26) {
				dp[i] = (dp[i] + dp[i - 2]) % 1000000;
			}
		}
		System.out.print(dp[N] % 1000000);
	}

}