[Algorithm] 🔐백준 2011 암호 코드

0. 문제

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

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

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

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

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

1. 아이디어

💡 0 처리
1) 첫번째 숫자가 0인 경우
2) 마지막 숫자가 0이고 마지막 직전의 숫자가 1,2가 아닐 때
3) 그 외 중간에 10,20이 아닌 0이 들어가는 경우

💡 현재 숫자가 1≤x≤9이면, dp[x] = dp[x-1]
바로 앞 숫자와 합해서 10≤x≤26이면, dp[x] = dp[x-1] + dp[x-2]
(0처리의 3번이 이 코드로 해결됨)

2. 핵심 풀이

0처리
1. 첫번째 숫자가 0인 경우 / 마지막 숫자가 0이고 마지막 직전의 숫자가 1,2가 아닐 때

if (str.charAt(0) == '0')
	flag = 1;
else if (str.charAt(length - 1) == '0' && str.charAt(length - 2) != '1' && str.charAt(length - 2) != '2')
	flag = 1;

2. 현재 숫자가 1≤x≤9이면,
   dp[x] = dp[x-1]
   바로 앞 숫자와 합해서 10≤x≤26이면,
   dp[x] = dp[x-1] + dp[x-2]

else {
	for (int i = 2; i <= length; i++) {
		if (str.charAt(i-1) > '0')
			dp[i] = dp[i - 1] % 1000000;

		int tmp = Integer.parseInt(str.substring(i-2,i));
		if (10 <= tmp && tmp <= 26)
			dp[i] = (dp[i] + dp[i - 2]) % 1000000;
	}
}

3. 코드

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

public class DP_15 {
	public static void main(String[] args) throws IOException {
		BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
		BufferedWriter bw = new BufferedWriter(new OutputStreamWriter(System.out));
		
		String str = br.readLine();
		int length = str.length();
		long[] dp = new long[length + 1];
		int flag = 0;
		
		dp[0] = dp[1] = 1;

		if (str.charAt(0) == '0')
			flag = 1;
		else if (str.charAt(length - 1) == '0' && str.charAt(length - 2) != '1' && str.charAt(length - 2) != '2')
			flag = 1;
		else {
			for (int i = 2; i <= length; i++) {
				if (str.charAt(i-1) > '0')
					dp[i] = dp[i - 1] % 1000000;

				int tmp = Integer.parseInt(str.substring(i-2,i));
				if (10 <= tmp && tmp <= 26)
					dp[i] = (dp[i] + dp[i - 2]) % 1000000;
			}
		}
		
		if(flag != 0)
			bw.write("0");
		else
			bw.write(dp[length]+"");
		
		bw.flush();
		bw.close();
	}
}

4. 결과

성공✨