[백준] 동적 계획법 - 2011번: 암호코드 (Java)

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

입력에 0이 포함되어있을 때 예외처리를 잘 해주면 된다.

입력 받은 문자열을 charAt메소드로 탐색해주면서

• 현재 0일 때
• 앞이 0일 때
• 뒤가 0일 때
• 0이 아닐 때로 나누어서 탐색해주었다.

알파벳은 26개이다. 두 문자를 합쳐서 1 ~ 26 범위에 있는지 체크해준다.

문제의 예시로 설명해보겠다.

2	5	1	1	4
1	2	2	4	6

위의 표는 해당 자리까지 탐색했을 때 몇가지 경우의 수가 나오는지 체크한 표이다.

2: { 2 } - 1가지

25: { { 2, 5 }, { 25 }} - 2가지

251: { { 2, 5, 1 }, { 25, 1 } } - 2가지

2511: { { 2, 5, 1, 1 }, { 25, 1, 1 }, { 2, 5, 11 }, { 25, 11 } } - 4가지

25114: { { 2, 5, 1, 1, 4 }, { 25, 1, 1, 4 }, { 2, 5, 11, 4 }, { 25, 11, 4 }, { 2, 5, 1, 14 }, { 25, 1, 14 } } - 6가지

예시는 0이 없어 간단하지만, 생각을 꽤 오래했다.

첫번째 문자는 0이 아니면 무조건 1개의 경우의 수가 나오므로 처음부터 dp[0] = 1로 초기화 한다.

두번째 문자부터 반복문을 시작한다.

규칙을 찾아보자면, 앞자리를 합쳤을 때 26이하일때와 아닐때를 나누어준다.

앞자리와 현재자리를 합쳐서 26보다

• 크면
  • 무조건 둘로 나누어야 한다. 예를 들어 이번에 탐색할 자리의 수가 7이고 앞자리가 3이다(...37...). 그렇다면 앞과 합치면 37이므로 37에 해당하는 알파벳은 없다. 3인 c, 7인 g로 무조건 나누어야 한다.
  • dp[i] = dp[i - 1]
• 작으면
  • ex) ....12....
  • 이때는 ... ab... 이렇게 될 수도 있지만, 12에 해당하는 l 이 들어갈 수도 있어서 ....l.... 이 될 수도 있다.

이런식으로 예외처리를 해주고 현재가 0, 앞이 0, 뒤가 0을 예외처리 해주면 된다.

그리고 dp 배열에 입력할 때 1000000으로 나눈 나머지 처리를 잊지 말자.