상근이와 선영이가 다른 사람들이 남매간의 대화를 듣는 것을 방지하기 위해서 대화를 서로 암호화 하기로 했다. 그래서 다음과 같은 대화를 했다.
어떤 암호가 주어졌을 때, 그 암호의 해석이 몇 가지가 나올 수 있는지 구하는 프로그램을 작성하시오.
첫째 줄에 5000자리 이하의 암호가 주어진다. 암호는 숫자로 이루어져 있다.
나올 수 있는 해석의 가짓수를 구하시오. 정답이 매우 클 수 있으므로, 1000000으로 나눈 나머지를 출력한다.
암호가 잘못되어 암호를 해석할 수 없는 경우에는 0을 출력한다.
25114
6
이 문제는 dp(다이나믹 ) 알고리즘을 이용해서 풀 수 있었다.
import java.io.BufferedReader;
import java.io.InputStreamReader;
public class Main {
public static void main(String[] args) throws Exception{
BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
String input = br.readLine();
if(input.equals("") || input.charAt(0)=='0') {
System.out.println(0);
return;
}
sol(input);
}
public static void sol(String str) {
long[][] dp = new long[str.length()+1][2];
dp[0][0] = 1;
dp[1][0] = 1;
for(int i=2; i<=str.length(); i++) {
int one = str.charAt(i-1) - '0';
int ten = str.charAt(i-2) - '0';
if(one==0) { //일의 자리 0인 경우
if(ten==0) {
System.out.println(0);
return;
}
else {
dp[i][0] = 0; //한자리 수 없는 경우
if(ten==1 || ten==2) { //두 자리 수 가능한 경우
dp[i][1] = (dp[i-2][0] + dp[i-2][1]) % 1000000;
}
}
}
else {
dp[i][0] = (dp[i-1][0]+dp[i-1][1]) % 1000000;
//-----일의 자리 있는 경우
if(ten==1 && (one>=0 && one<=9)) { //십의 자리가 1일 때는 일의 자리 0~9 가능
dp[i][1] = (dp[i-2][0] + dp[i-2][1]) % 1000000;
}
else if(ten==2 && (one>=0 && one<=6)) { //십의 자리가 2일 때는 일의자리 0~6까지 가능
dp[i][1] = (dp[i-2][0] + dp[i-2][1]) % 1000000;
}
else
dp[i][1] = 0;
}
}
System.out.println((dp[str.length()][0]+dp[str.length()][1]) % 1000000);
}
}