문제
상근이와 선영이가 다른 사람들이 남매간의 대화를 듣는 것을 방지하기 위해서 대화를 서로 암호화 하기로 했다. 그래서 다음과 같은 대화를 했다.
- 상근: 그냥 간단히 암호화 하자. A를 1이라고 하고, B는 2로, 그리고 Z는 26으로 하는거야.
- 선영: 그럼 안돼. 만약, "BEAN"을 암호화하면 25114가 나오는데, 이걸 다시 글자로 바꾸는 방법은 여러 가지가 있어.
- 상근: 그렇네. 25114를 다시 영어로 바꾸면, "BEAAD", "YAAD", "YAN", "YKD", "BEKD", "BEAN" 총 6가지가 나오는데, BEAN이 맞는 단어라는건 쉽게 알수 있잖아?
- 선영: 예가 적절하지 않았네 ㅠㅠ 만약 내가 500자리 글자를 암호화 했다고 해봐. 그 때는 나올 수 있는 해석이 정말 많은데, 그걸 언제 다해봐?
- 상근: 얼마나 많은데?
- 선영: 구해보자!
어떤 암호가 주어졌을 때, 그 암호의 해석이 몇 가지가 나올 수 있는지 구하는 프로그램을 작성하시오.
입력
첫째 줄에 5000자리 이하의 암호가 주어진다. 암호는 숫자로 이루어져 있다.
출력
나올 수 있는 해석의 가짓수를 구하시오. 정답이 매우 클 수 있으므로, 1000000으로 나눈 나머지를 출력한다.
암호가 잘못되어 암호를 해석할 수 없는 경우에는 0을 출력한다.
나의 풀이
이 문제는 DP로 해결해야 하는 문제이다. 문제에서 생각해야 할 것은 두 숫자를 합쳤을 때 그 숫자가 1부터 26사이에 들어가는지를 고려하고 0이 들어갔을 때를 고려하면 해결할 수 있다.
fun main() = with(System.`in`.bufferedReader()) {
val n = readLine()
if (n.first() == '0') {
println(0)
return@with
}
val dp = IntArray(n.length + 1)
dp[0] = 1
dp[1] = 1
for (i in 2 until dp.size) {
if(n[i - 1] == '0'){
if(n.substring(i - 2, i).toInt() in 1.. 26){
dp[i] = (dp[i] + dp[i - 2]) % 1000000
}
}
else{
dp[i] = (dp[i] + dp[i - 1]) % 1000000
if (n.substring(i - 2, i).toInt() in 11..26) {
dp[i] = (dp[i] + dp[i - 2]) % 1000000
}
}
}
println(dp.last())
}
'백준 > DP' 카테고리의 다른 글
12825번: 1로 만들기 2 - Kotlin, Java (1) | 2023.10.05 |
---|---|
9184번: 신나는 함수 실행 - Kotlin, Java (0) | 2023.10.01 |
2096번: 내려가기 - Kotlin (0) | 2023.06.13 |
2565번: 전깃줄 - Kotlin (0) | 2023.06.11 |
2225번: 합분해 - Kotlin (0) | 2023.06.07 |