[Python] 백준 2011번 - 암호 코드 (골드 5)

혼자 힘으로 풀었는가? O

알고리즘 분류
 - 다이나믹 프로그래밍(DP)

 

https://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을 출력한다.

 

---

A부터 Z를 1~26으로 표기할 때 가능한 경우의 수를 찾는 문제이다.

 

여기서 중요한 점은 0에대한 관리이다.

 

10, 20은 J, T 로 매칭이 가능하지만

 

00, 30, 40 ~ 등은 존재하지 않는다.

 

또한 첫 글자가 0일 경우에는 조합이 불가능하다. (01, 021, 0443 등)

 

DP점화식을찾는 법은

 

예제 2의 1111111111 을 보면 쉽다.

 

 

위 처럼 dp[i] = dp[i-1] + dp[i-2] 임을 알 수 있다.

 

앞의 숫자가 큰 경우에는

 

dp[i] = dp[i-1] 임을 알 수 있다.

 

 

import sys

input = sys.stdin.readline

data = list(input().rstrip())
n = len(data)
if data[0] == '0': # 첫 글자가 0
    print(0)
else:
    dp = [0] * (n+1)
    dp[0] = 1
    dp[1] = 1

    for i in range(2, n+1):
        if data[i-1] == '0': # 현재 글자가 0
            if data[i-2] == '0' or int(data[i-2]) > 2: # 그 앞 글자가 0 혹은 2 초과
                dp[n] = 0
                break
            else:
                dp[i] = dp[i-2]
        elif data[i-2] == '0': # 앞 글자가 0일 때
            dp[i] = dp[i-1]
        elif int(data[i-2] + data[i-1]) <= 26: # 1~26 사이
            dp[i] = dp[i-1] + dp[i-2]
        else: # 26 초과
            dp[i] = dp[i-1]
        dp[i] %= 1000000
            
    print(dp[n])