-
백준 2011번 암호코드(자바) 풀이 - DP코딩테스트 2021. 9. 8. 18:07
암호코드
한국어
시간 제한메모리 제한제출정답맞은 사람정답 비율
2 초 128 MB 34772 6173 4510 19.593% 문제
상근이와 선영이가 다른 사람들이 남매간의 대화를 듣는 것을 방지하기 위해서 대화를 서로 암호화 하기로 했다. 그래서 다음과 같은 대화를 했다.
- 상근: 그냥 간단히 암호화 하자. A를 1이라고 하고, B는 2로, 그리고 Z는 26으로 하는거야.
- 선영: 그럼 안돼. 만약, "BEAN"을 암호화하면 25114가 나오는데, 이걸 다시 글자로 바꾸는 방법은 여러 가지가 있어.
- 상근: 그렇네. 25114를 다시 영어로 바꾸면, "BEAAD", "YAAD", "YAN", "YKD", "BEKD", "BEAN" 총 6가지가 나오는데, BEAN이 맞는 단어라는건 쉽게 알수 있잖아?
- 선영: 예가 적절하지 않았네 ㅠㅠ 만약 내가 500자리 글자를 암호화 했다고 해봐. 그 때는 나올 수 있는 해석이 정말 많은데, 그걸 언제 다해봐?
- 상근: 얼마나 많은데?
- 선영: 구해보자!
어떤 암호가 주어졌을 때, 그 암호의 해석이 몇 가지가 나올 수 있는지 구하는 프로그램을 작성하시오.
입력
첫째 줄에 5000자리 이하의 암호가 주어진다. 암호는 숫자로 이루어져 있다.
출력
나올 수 있는 해석의 가짓수를 구하시오. 정답이 매우 클 수 있으므로, 1000000으로 나눈 나머지를 출력한다.
암호가 잘못되어 암호를 해석할 수 없는 경우에는 0을 출력한다.
예제 입력 1 복사
25114
예제 출력 1 복사
6
이 문제는 어렵지는 않았지만 예외로 생각해야 할 것들이 있어서 조금 헷갈렸다.
또 구현하는데 오류가 발생해서 오래걸렸다
풀이 : 예시로 나온 25114를 살펴보자.
2일 때 - 2
25일 때 - 2,5 / 25
251일 때 - 2,5,1 / 25,1 /
2,51(이 경우는 51로 만들 수 있는 문자가 없으므로 제외)2511일 때 - 2,5,1,1 / 25,1,1 / 2,5,11 / 25,11
25114일 때 - 2,5,1,1,4 / 25,1,1,4 / 2,5,11,4 / 25,11,4 / 2,5,1,14 / 25,1,14
즉, 25114일 때는 2511일 때 경우에 4를 추가 해준 것과 251일 때 경우에 14를 추가 해준 것을 합친 것과 같다.
이처럼 dp[n] = dp[n-1] + dp[n-2] 임을 알 수 있다.
예외 :
하지만 251일 때를 생각해보면 25일 때의 경우에 1을 추가해준 것과 2일 때의 경우에 51을 추가 해준 것을 합쳐야 하는데 51은 암호화 할 수 없으므로 dp[n] = dp[n-1]이다.
0이 들어가는 경우 0의 앞에 반드시 1이나 2가 와야 하고, 그렇지 않은 경우에는 잘못된 암호이므로 0을 출력한다.
0 앞에 1이나 2가 온 경우는 예를 들어 2510일 때, 25일 때의 경우에 10을 추가해주는 경우만 가능하므로
dp[n] = dp[n-2]이다.
0 뒤에 숫자가 올 때도 고려해야 한다.
예를 들어 25104를 생각해보면 2510일 때의 경우에 4를 추가하는 경우와 251에 04를 추가해준 경우로 나눌 수 있는데 04는 암호화가 되지 못하므로 2510에 4를 추가하는 경우만 유효하다.
따라서 dp[n] = dp[n-1]이다.
import java.io.BufferedReader; import java.io.InputStreamReader; import java.util.StringTokenizer; public class Main { public static void main(String args[]) throws Exception{ BufferedReader br = new BufferedReader(new InputStreamReader(System.in)); StringTokenizer st = new StringTokenizer(br.readLine()); String str = st.nextToken(); long[] dp = new long[str.length()+1]; if (str.charAt(0) == '0') { //0입력하면 System.out.println(0); return; } else { dp[0] = 1; dp[1] = 1; for (int i = 2; i <= str.length(); i++) { if (str.charAt(i-1) == '0') { if (str.charAt(i-2) == '1' || str.charAt(i-2) == '2') dp[i] = dp[i-2] % 1000000; else break; } else { int result = Integer.parseInt(str.substring(i-2,i)); if (result < 27 && result > 9) dp[i] = (dp[i-1] + dp[i-2]) % 1000000; else { dp[i] = dp[i-1] % 1000000; } } } } System.out.println(dp[str.length()] % 1000000); } }
str.charAt(i-1) == '0'으로 쓰지 않고 (str.charAt(i-1) - '0') < 1 이렇게 써서 오류가 났다.
'코딩테스트' 카테고리의 다른 글
백준 11650 좌표 정렬하기(자바) - 정렬(Comparator) (0) 2021.09.30 백준 11052번 카드 구매하기(자바) - DP (0) 2021.09.30 백준 9461번 파도반 수열(자바) - DP (0) 2021.09.08 **백준 2133번 타일 채우기(자바) - DP (0) 2021.08.09 *백준 1699번 제곱수의 합(자바) - DP (0) 2021.07.29