문제

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

  • 상근: 그냥 간단히 암호화 하자. A를 1이라고 하고, B는 2로, 그리고 Z는 26으로 하는거야.

  • 선영: 그럼 안돼. 만약, “BEAN”을 암호화하면 25114가 나오는데, 이걸 다시 글자로 바꾸는 방법은 여러 가지가 있어.

  • 상근: 그렇네. 25114를 다시 영어로 바꾸면, “BEAAD”, “YAAD”, “YAN”, “YKD”, “BEKD”, “BEAN” 총 6가지가 나오는데, BEAN이 맞는 단어라는건 쉽게 알수 있잖아?

  • 선영: 예가 적절하지 않았네 ㅠㅠ 만약 내가 500자리 글자를 암호화 했다고 해봐. 그 때는 나올 수 있는 해석이 정말 많은데, 그걸 언제 다해봐?

  • 상근: 얼마나 많은데?

  • 선영: 구해보자!

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

www.acmicpc.net/problem/2011

  • 입력

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

  • 출력

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

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

최종 코드

하다가 모르겠으면 답을 봐야하는데 오기가 생겨서 1시간 넘게 붙잡고 있다가 결국 답을 확인했다.

예를 들어 25114라는 5자리수 암호를 해석하는 경우의 수는

dp[3]은 (2, 5, 1), (25, 1)로 2가지, dp[4]는 (2, 5, 1, 1), (25, 1, 1), (2, 5, 11), (25, 11, 4)로 4가지이기 때문에 dp[5]는 (2, 5, 1, 1, 4), (25, 1, 1, 4), (2, 5, 11, 4), (25, 11, 4, 4)로 dp[4] 뒤에 4를 추가하는 4가지 경우에 (2, 5, 1, 14), (25, 1, 14)로 dp[3] 뒤에 14를 추가하는 2가지 경우를 더해서 계산해야 한다.

따라서 지금 구하는 마지막 한 자리 숫자가 0이 아니면 dp[i-1]을 더하고, 마지막 두 자리 숫자가 1~26 사이면 dp[i-2]도 더해주어야 하는 것이다.

이를 코드로 구현하면 아래와 같다.

s = input()
dp = [1, 1]
cnt = 0

if int(s[0]) == 0:
    print(0)
else:
    for i in range(1, len(s)):
        if int(s[i]) != 0:
            cnt += dp[-1]
        if 10 <= int(s[i-1:i+1]) <= 26:
            cnt += dp[-2]
        elif int(s[i-1:i+1]) == 0:
            dp[-1] = 0
            break
        dp[-2] = dp[-1]
        dp[-1] = cnt
        cnt = 0

    print(dp[-1]%1000000)

참고 사이트