[백준 2011 - C++] 암호코드 : DP

www.acmicpc.net/problem/2011

2011번: 암호코드

문제

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

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

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

 

입력

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

 

출력

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

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

 

예제 입력 1 

25114

예제 출력 1 

6

 

풀이

다 풀고나서 든 생각은 굳이 2차원배열을 쓸 필요가 없는데 써서 공간낭비를 좀 심하게 했다는 것...ㅋㅋㅋㅋㅋ

끝숫자가 하나인 경우엔 dp[n-1]로, 두개인 경우엔 dp[n-2]로 간단하게 할 수 있다...

풀 때는 왜 생각이 안나나 몰라...

이 문제는 '0'을 처리하는게 아주 중요한 문제다!

특히 맨앞에 '0'이 온 경우와 '0' 두 개가 연달아서 나타나는 경우

 

#include <iostream>
#include <cstdio>
#include <string.h>

using namespace std;

char pw[5001];
int can[5001][2];

int main (void){
    int len;
    int total;
    
    scanf("%s", pw);
    
    if(pw[0] == '0'){
        printf("0\n");
        return 0;
    }
    
    can[0][0] = 1;
    can[0][1] = 0;
    len = strlen(pw);
    
    for(int i=1; i<len; i++){
        //printf("%d %d\n", can[i-1][0], can[i-1][1]);
        if(pw[i] == '0'){
            if(pw[i-1] == '1' || pw[i-1] == '2'){
                can[i][0] = 0;
                can[i][1] = can[i-1][0]%1000000;
                continue;
            }
            else{
                printf("0\n");
                return 0;
            }
                
        }
        else if(pw[i-1] == '1' && pw[i] >= '0' && pw[i] <= '9'){
            can[i][1] = can[i-1][0]%1000000;
        }
        else if(pw[i-1] == '2' && pw[i] >= '0' && pw[i] <= '6'){
            can[i][1] = can[i-1][0]%1000000;
        }
        can[i][0] = (can[i-1][0] + can[i-1][1])%1000000;
    }
    
    total = (can[len-1][0] + can[len-1][1])%1000000;
    printf("%d\n", total);
}