[🥈1 / 백준 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  1111111111	예제 출력 2  89

 

풀이

 

숫자 리스트를 알파벳으로 변환할 때 만들 수 있는 가짓수를 동적 계획법으로 구하는 문제이다.

dp 배열에는 각 자리수까지 수를 사용할 때 만들 수 있는 가짓수를 담는다.

개인적으로 방법을 떠올리기가 쉽지 않았다.

 

앞,뒤 수를 합쳐 26이하의 수를 구성할 수 있다면 합칠 수 있다.

252525 같은 수를 보자면 총 3군데에서 합칠 수 있게 되는데, 2^3 경우의 수가 존재하게 된다.

즉, 앞뒤로 합칠 수 있다면 2의 배수로 경우의 수가 증가하는 것이다.

 

단 111과 같이 연속적으로 합칠 수 있는 구간이 존재한다면 직전의 dp값에 2배수가 아닌, 피보나치 형식으로 증가하게 된다.

즉, dp[i] = dp[i-2]+dp[i-1]이 되는 것이다.

 

두 가지를 염두하여 코드를 작성해야 한다.

또한, 0에 대한 예외처리가 필요하다.

A = 1 이므로 0이 단독으로 있어야 하는 경우 암호를 해독할 수 없다.

 

import sys
input = sys.stdin.readline

s = list(map(int,list(input().rstrip())))
dp = [1] + [0]*(len(s)-1)
# 직전에 합쳐졌었는지를 comb boolean값으로 구분했다.
comb = False
if s[0] == 0:
    print(0)
else:
    for i in range(1,len(s)):
        if s[i] > 0:
            dp[i] = dp[i-1]%1000000
        if s[i-1] == 1 or (s[i-1] == 2 and s[i] <= 6):
            if comb:
                dp[i]+=dp[i-2]
            else:
                dp[i]+=dp[i-1]
            comb = True
        else:
            comb = False
    print(dp[-1]%1000000)