[백준] 2011 암호코드

문제

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

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

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

입력

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

출력

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

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

 

여러가지 엣지케이스를 조심해야합니다.

0을 중심으로 생각해야합니다.

0이 들어오면 그 수는 무조건 앞수와 붙어서 사용해야하므로 a[i]=a[i-2]인데, 그 앞수가 2보다 크면 안됩니다.(30)

 

0이 아닌숫자가 들어오면 a[i]=a[i-1]+조건부a[i-2]입니다. 이때 s[i-1]이 0이 아니어야 a[i-2]를 더할 수 있습니다.

 

0으로 시작하는 string이 들어오면 무조건 0입니다.

 

또한 00, 000처럼 0이 2개이상연속해서 들어와도 0입니다.

 

	#include<bits/stdc++.h>
	#define FAST ios_base::sync_with_stdio(false),cin.tie(NULL);
	#define mset(v) memset(v,0,sizeof(v));
	#define rep(i,a) for(int i=0;i<a;++i)
	#define REP(i,a) for(int i=1;i<=a;++i)

	using namespace std;

	typedef long long ll;
	typedef pair<int, int> pi;
	typedef tuple<int, int, int>ti;
	typedef vector<int> vi;
	typedef vector<vector<int>> vvi;
	int dy[4] = { -1,0,1,0 }, dx[4] = { 0,1,0,-1 };

	int n, a[5001], s[5001];
	int main() {
		FAST;
		char c;
		int num = 0;
		bool z = 0;
		while (cin >> c) {
			s[++num] = c - '0';
		}
		a[0] = 1;
		a[1]= 1;
		if (s[1] == 0) {
			cout << 0;
			return 0;
		}
		for (int i = 2; i <= num; i++) {
			if (s[i] != 0) {
				z = 0;
				a[i] += a[i - 1];
				a[i] %= 1000000;
				if (s[i-1]!=0 && (s[i - 1] * 10 + s[i]) <= 26) {
					a[i] += a[i - 2];
				}
			}
			else {
				if (z || s[i - 1] > 2) {
					cout << 0;
					return 0;
				}
				z = 1;
				a[i] += a[i - 2];
			}
			a[i] %= 1000000;
		}
		cout << a[num] % 1000000;
	}