[BOJ] 2011 - 암호코드

문제

문제

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

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

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

입력

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

출력

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

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

예제 입력

25114

예제 출력

6

문제 해결

우선 각 단계의 가지수를 dp 배열로 잡고 생각해보자. 각 단계의 가지수를 생각해보자. 우선 예제로 주어진 25114를 가지고 경우를 찾아보자.

걸러진 암호는 가능한 암호이므로 dp[0] = 1

첫번째 수인 2는 암호 코드 경우가 하나이다. 따라서 dp[1] = 1

두번째 수인 5는 우선 i-1 (1) 단계의 경우의 수 dp[2] 에 더한다. dp[2] += dp[1]

그리고 1번째와 2번째를 이어붙인 25 또한 암호 코드에 속하므로 dp[2]에 dp[i-2] (0) 단계의 경우의 수를 더해야 한다. dp[2] += dp[0]

위 두단계를 통해 알아낸 규칙을 정리해보자

해당 단계 숫자가 0 ~ 9 인 경우

dp[i] += dp[i-1]

해당 단계 숫자와 전단계 숫자가 붙혔을때 10 ~ 24 인 경우

dp[i] += dp[i-2]

불가능한 숫자 암호

0으로 시작하는 암호는 문제에서 말하는 잘못된 암호로 판단하여 0을 출력후 종료

python 코드

a = list(map(int, list(input())))
l = len(a)
dp = [0] * (l+1)
if a[0] == 0:
    print(0)
else :
    a = [0] + a # 인덱싱을 위해 추가한 0
    dp[0] = 1
    dp[1] = 1

    for i in range(2, l+1):
        cur = a[i]
        cur2 = a[i-1] * 10 + a[i]
        if cur >= 1 and cur <= 10:
            dp[i] += dp[i-1]
        if cur2 >= 10 and cur2 <= 26:
            dp[i] += dp[i-2]
        dp[i] %= 1000000


    print(dp[l])

느낀점

오랜만에 다시 알고리즘을 풀었다. python 문법을 많이 까먹고 코드가 간결하지가 않다. 꾸준히 해야겠다.