[백준 2011] 암호코드 - Java 문제풀이

오늘의 1일 1골드 문제는 다음 문제이다.

 

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자리 이하의 암호가 주어진다. 암호는 숫자로 이루어져 있다.

해설

이 문제는 경우의 수를 나누어 생각하면 편하다.

 

1. 암호가 잘못되어 해석할 수 없는 경우
2. 해당 자리 숫자가 1의 자리인 한 자리 수
3. 해당 자리 숫자가 1의 자리인 두 자리 수

먼저 1번의 경우, 암호를 해석할 수 없는 경우는 첫번째 암호가 0인 경우에만 해당하기 때문에 이 경우만 확인하고 넘어가면 된다.

 

2번 해당 자리 숫자가 1의 자리인 한 자리 수인 경우는 해당 자리 숫자가 0인 경우만 체크해주면 된다. 1~9까지는 모두 A~I까지의 알파벳이 대응되기 때문이다. 그러한 경우에는, arr[i] += arr[i-1]가 되는데, 그 이유는 이전까지 가능한 경우에 길이가 +1 되는 것이기 때문이다.

 

3번 해당 자리 숫자가 1의 자리인 두 자리 수인 경우도 위와 마찬가지로 이전+현재의 숫자가 10 ~ 26 이하인 경우만 체크해주면 된다. 만약 가능하다면 arr[i] += arr[i-2]가 되는데, 2자리수 전까지 가능하다가, +2 만큼 길이가 늘어나기 때문이다.

 

 

그렇다면 전제조건을 생각해보자. 2, 3번을 확인하면 -2까지 체크해줌을 알 수 있다. 따라서 전제조건으로 문자열 길이가 1 이상인 경우와 2이상인 경우를 나누어 생각한다.

 

문자열 길이가 1이상인 경우에는 무조건 +1 해준다. 그 이유는 어차피 1번을 확인하고 넘어왔기 때문에 주어진 하나의 수로 만들 수 있는 길이 1의 문자는 1개이기 때문이다.

 

문자열 길이가 2이상인 경우는 위 2, 3번을 체크해주면 된다. 1~9인 경우에 +1, 첫번째 수와 합쳐 2자리 수를 만들었을 때 26 이하인 경우 +1을 해주면 된다.

 

다음은 코드 전문이다.

import java.util.Scanner;

public class BOJ_2011 {
    public static void main(String[] args) {
        Scanner sc = new Scanner(System.in);
        String str = sc.next();
        int N=str.length();
        int[] arr = new int[N];
        for(int i=0;i<N;i++) arr[i] = str.charAt(i)-'0';
        System.out.println(cal(arr, N));
    }
    static int cal(int[] arr, int N) {
        if (arr[0] == 0) return 0;
        int[] answer = new int[N];
        if (N>=1) answer[0] = 1;
        if (N>=2) {
            if (arr[1]!=0) answer[1] =1;
            if (isValid(arr[0], arr[1])) answer[1]++;
        }
        for(int i=2;i<N;i++) {
            if (arr[i]!=0) answer[i]=answer[i-1];
            if (isValid(arr[i-1], arr[i])) answer[i] += answer[i-2];
            answer[i]%=1000000;
        }
        return answer[N-1];
    }
    static boolean isValid(int before, int now) {
        if (before == 0) return false;
        else if (before*10 + now <= 26) return true;
        else return false;
    }
}

 

 

 

---

 

DP를 계속 풀며 느낀게 전제조건과 경우의 수만 잘 따지면 된다는 것이다. 이 문제는 계속 왔다갔다 하면서 푸느라 30분정도가 걸렸다. 자리에 계속 앉아서 풀었으면 20분 내외로 풀렸을 듯 싶다.

DP.. 계속 연습해야 겠다. ㅎ