본문 바로가기

이론/문제풀이

[백준 알고리즘 -2011] 암호코드 - java (신나는 삽질일기)

문제만 보면 여느 DP 문제인데 정답률이 왜이렇게 낮나 했다.

제가 그 낮은 정답률에 또 일조하고 왔슴다 (_ _)

문제


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

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

평범한 DP문제처럼 이 문제도 접근하면 된다.

OOOOOOOOOO ... OOOO

이렇게 문자열이 주어졌을 때,

DP[N] = N번째 문자까지 나올 수 있는 암호 해석의 수

로 해두면, 점화식은 두가지로 나뉘어진다.

  1. N번째 문자까지 두자리가 문자 하나를 가리키는 경우

-> 10~26까지의 알파벳 때문에 가능하다.

  1. 한자리 문자인 경우

식으로 세워보면,

DP[N] = DP[N-1] (if: 한자리 문자만 가능한 경우)
DP[N] = DP[N-1] + DP[N-2] (if: 두자리 문자도 가능한 경우)

여기서 더 쪼개면,

두자리 문자만 가능한 경우까지 나올 수 있지만 그렇게까지 나누지 않고도 코딩 가능하다.

이 문제가 정답률이 작살난 이유는 난이도가 아니라 ㅠ__ㅠ

'0' 처리 때문이다.

"암호가 잘못되어 암호를 해석할 수 없는 경우에는 0을 출력한다." 이런 조건은 읽어도 가볍게 넘기게 되는 것 같다. 반성

결론은 0으로 시작하는 경우에 0을 출력해줘야하고,

1230 처럼, 맨 마지막이 0인데 두글자가 안되는 경우도 0을 출력해주어야한다.

코드,

import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;


public class Main {

   public static void main(String[] args) throws IOException {

       BufferedReader br=new BufferedReader(new InputStreamReader(System.in));
       String N= br.readLine();

       long[] dp=new long[N.length()+1];
       dp[0]=1;
       dp[1]=1;

       if(N.charAt(0)=='0'){//0으로 시작하는 경우
           System.out.println(0);
           return;
      }

       for(int i=1; i<N.length(); i++) {
           char pri = N.charAt(i - 1);//이전 숫자
           if(N.charAt(i) >= '1' && N.charAt(i)<='9'){//혼자올수있음
               dp[i+1]+=dp[i];
               dp[i+1]%=1000000;
          }
           if (!(pri == '0' || pri > '2' || (pri == '2' && N.charAt(i) > '6'))) {
               //두글자가 될 수 있다면
               dp[i + 1] += dp[i-1];
               dp[i+1]%=1000000;
          }
           dp[i + 1] %= 1000000000;
           /*이렇게 처리하면,
           1230 처럼, 맨 마지막이 0인데 두글자가 안되는 경우도 0을 출력해주어야한다.
           는 자동으로 처리된다.
           */

      }

       System.out.println(dp[N.length()]%1000000000);

  }
}