[백준/골드5] 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 

1111111111

예제 출력 2 

89

 

 

풀이

25114라는 수가 주어졌을 때, 문제를 왼쪽부터 처리한다고 가정하면

2

25 : 2/5 , 25 -> 2가지 경우 존재

251: 2/51 (51>26이므로 불가능) ,  25/1 -> 1가지 경우 존재

이전에 나누어 놓은 상태가 다음 상태에 영향을 미친다. 따라서 DP로 해결한다.

dp[i] : 앞에서부터 i개의 문자를 해석할 수 있는 경우의 수

해석할 수 있는 조건 = 1자리 수, 2자리수(10<=num<=26)

 

① 1자리 수로 처리

i번째 수를 1자리 수로 처리하면, 경우의 수는 이전 자리(i-1)에 처리한 경우의 수와 같다.

dp[i] = dp[i-1]

 

② 2자리 수로 처리

i번째 수를 2자리 수로 처리하면, i-1과 함께 처리해야하므로 i-2번째 자리에서 처리한 결과값에 영향을 받는다.

dp[i] = dp[i-2]

이때, 2자리 수는 범위가 10<= num <=26이여야한다.

 

*예외 처리

문제에서 주의할 점으로, 암호가 잘못되어 해석할 수 없는 경우는 0으로 처리한다고 하였다.

1) 한 자리 수로 처리할 때, 0을 처리할 수 없다.

2) 0으로 시작하는 암호는 처리할 수 없다.

 

*초기화

i-2가 점화식에 들어가므로, 0,1인덱스의 값을 초깃값으로 정의해주어야한다.

 

dp[0] = 앞에서부터 0개를 사용 => 빈 문자열

이 문자열 뒤에 1자리 수 or 2자리수를 붙여나갈 수 있으므로, dp[0] = 1

따라서 dp[1] = 1

 

 

 

코드

import sys

code = sys.stdin.readline().replace(" ","")
n = len(code)
#dp[i] : 앞에서부터 i개 사용했을 때, 해석할 수 있는 경우의 수
dp = [0]*(n+1) #_25114 
dp[0] = 1 
dp[1] = 1

if code[0] == '0':
    print(0)
    exit()


for i in range(2,n+1):
    #한 자리
    if code[i-1] != '0':
        dp[i] += dp[i-1]
    
    #두 자리
    num = int(code[i-2:i])
    if(10<= num <= 26):
        dp[i] += dp[i-2]
    
    dp[i]%=1000000

print(dp[n])