[백준 2011번] 파이썬 - 암호 코드

백준 2011 - 암호코드

시간 제한 2초, 메모리 제한 128MB

# 조건

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

입력

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

출력

• 나올 수 있는 해석의 가짓수를 구하시오.
• 정답이 매우 클 수 있으므로, 1000000으로 나눈 나머지를 출력한다.
• 암호가 잘못되어 암호를 해석할 수 없는 경우에는 0을 출력한다.

# 접근 방법

• 우선 1~26의 범위의 숫자만 알파벳으로 변경할 수 있다.
• dp 상향식 접근을 이용하여 문제를 풀어주면 되는데 핵심은 1번 인덱스부터 차례로 몇 개의 암호 가짓수를 가지는지 구해주면 된다.
• 방법은 아래와 같다.
  • 현재 자리 숫자가 0보다 클 때 이전 dp값을 더해준다.
  • 이전 자리수와 현재 자리수를 두 자리수로 볼 때 10~26 사이의 숫자이면 2번째 전의 dp값 더해준다.
• 주의할 점은 0으로 시작하는 것은 예외처리를 해주어야 한다.

import sys  
sys.stdin = open('input.txt')  
input = sys.stdin.readline  

password = [0] + list(map(int, input().strip()))  
if password[1] == 0:  
    print(0)  
else:  
    leng = len(password)  
    can_make = [0] * leng  
    can_make[1] = 1  
    can_make[0] = 1  
    for i in range(2, leng):  
        val = password[i]  
        check = password[i-1] * 10 + password[i]  
        if val > 0:  
            can_make[i] += can_make[i-1]  
        if 10<=check<=26:  
            can_make[i] += can_make[i-2]  
    print(can_make[-1]%1000000)