출처 BOJ

암호코드

문제

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

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

  입력

  첫째 줄에 5000자리 이하의 암호가 주어진다.

  출력

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

  예제

  

범위체크

%$10^{6}$한 값끼리 더하는 거라 $\le INT$

풀이

암호의 i번째 값이 한자리일 경우, 두자리일 경우를 나누어 생각

• 한자리일 경우(1~9)
  이전까지 센 경우의 수와 동일 → sum[i] += sum[i-1]
  
• 두자리일 경우(10~26)
  i-1과 i가 한 set가 되므로 i-2번까지 센 경우의 수가 됨 → sum[i] += sum[i-2]
  

두가지 경우를 합친 값이 i번째의 경우의 수가 됨

코드

code

#include<cstdio>
#include<string>
#include<iostream>
#define DIV 1000000
using namespace std;

string code;
int sum[5002];

void count(){
	for(int i=1; i<code.size(); i++){
		if(code[i]-'0' >= 1)
			sum[i] = (sum[i] + sum[i-1]) % DIV;

		int temp = (code[i-1]-'0')*10 + (code[i]-'0');
	
		if(10 <= temp && temp < 27)
			sum[i] = (sum[i] + sum[i-2]) % DIV;
	}
}

int main(){
	cin>>code;
	sum[0] = 1;
	code = ' '+code;

	count();

	printf("%d\n",sum[code.size()-1]);
}

복잡도

$O(N)$
암호코드의 인덱스 0부터 길이 N까지 탐색