※주의※
저의 풀이가 정답은 아닙니다.
다른 코드가 더 효율적이거나 좋을 수 있습니다.
언제나 다른 사람의 코드는 참고만 하시기 바랍니다.
[문제 풀이 사이트]
2011번: 암호코드
나올 수 있는 해석의 가짓수를 구하시오. 정답이 매우 클 수 있으므로, 1000000으로 나눈 나머지를 출력한다. 암호가 잘못되어 암호를 해석할 수 없는 경우에는 0을 출력한다.
www.acmicpc.net
[문제 설명]
상근이와 선영이가 다른 사람들이 남매간의 대화를 듣는 것을 방지하기 위해서 대화를 서로 암호화 하기로 했다. 그래서 다음과 같은 대화를 했다.
- 상근: 그냥 간단히 암호화 하자. A를 1이라고 하고, B는 2로, 그리고 Z는 26으로 하는거야.
- 선영: 그럼 안돼. 만약, "BEAN"을 암호화하면 25114가 나오는데, 이걸 다시 글자로 바꾸는 방법은 여러 가지가 있어.
- 상근: 그렇네. 25114를 다시 영어로 바꾸면, "BEAAD", "YAAD", "YAN", "YKD", "BEKD", "BEAN" 총 6가지가 나오는데, BEAN이 맞는 단어라는건 쉽게 알수 있잖아?
- 선영: 예가 적절하지 않았네 ㅠㅠ 만약 내가 500자리 글자를 암호화 했다고 해봐. 그 때는 나올 수 있는 해석이 정말 많은데, 그걸 언제 다해봐?
- 상근: 얼마나 많은데?
- 선영: 구해보자!
어떤 암호가 주어졌을 때, 그 암호의 해석이 몇 가지가 나올 수 있는지 구하는 프로그램을 작성하시오.
[입력]
첫째 줄에 5000자리 이하의 암호가 주어진다. 암호는 숫자로 이루어져 있다.
[출력]
나올 수 있는 해석의 가짓수를 구하시오. 정답이 매우 클 수 있으므로, 1000000으로 나눈 나머지를 출력한다.
암호가 잘못되어 암호를 해석할 수 없는 경우에는 0을 출력한다.
[입출력 예]
입력 | 출력 |
25114 | 6 |
1111111111 | 89 |
[문제 풀이]
다이나믹 프로그래밍 문제이다.
0을 처리하는 부분 때문에 한번에 맞추진 못했다.
1. 테이블을 정의한다.
dp[n] = 길이에 따른 암호 코드 수, 인덱스가 길이를 의미한다.
2. 점화식을 정의한다.
점화식을 찾기에 앞서,
맨 첫번째가 0일 경우(ex) 07, 01, 05 등)는 암호코드를 만들 수 없기 때문에 0을 바로 출력해준다.
이제 예제로 나온 25114를 통해 점화식을 구해보자.
dp[1] (길이가 1일 때) = 1가지 (2)
dp[2] (길이가 2일 때) = 2가지 ((2, 5) , (25))
dp[3] (길이가 3일 때) = 2가지 ((2, 5, 1), (25, 1))
dp[4] (길이가 4일 때) = 4가지 ((2, 5, 1, 1), (2, 5, 11), (25, 1, 1), (25, 11))
dp[5] (길이가 5일 때) = 6가지 ((2, 5, 1, 1, 4), (2, 5, 11, 4), (2, 5, 1, 14), (25, 1, 1, 4), (25, 1, 14), (25, 11, 4))
그래서 아래와 같은 점화식을 구할 수 있다.
↪ dp[n] = dp[n - 1] + dp[n - 2]
하지만 여기서 끝난게 아니다. 입력 중간에 0이 존재하는 경우가 있다.
예를 들면) 110, 12204 .. 등
110의 경우 ((1, 10)) 1가지이다.
그래서 현재 값이 0인 경우 이전 값을 확인한다.
이전 값이 1 또는 2인 경우 dp[n] = dp[n - 2] 이다.
이전 값이 1 또는 2이 아닌 경우(ex) 130 .. 등) 암호코드를 만들 수 없기 때문에 break를 통해 0을 출력한다.
12204의 경우 ((1, 2, 20, 4), (12, 20, 4)) 2가지이다.
4를 확인할 때 이전 값인 0을 확인하고, 총 합을 구한다 (총 합 : 0 * 10 + 4 = 4)
총 합이 11 ~ 26이 아니라면 dp[n] = dp[n - 1] 이다.
총 합이 11 ~ 26 이면 dp[n] = dp[n - 1] + dp[n - 2] 이다.
0을 처리하는 경우까지 생각해서 코드를 작성하면 된다.
3. 초기값을 정의한다.
예제를 통해 dp[0] = 1, dp[1] = 1 초기 값을 구할 수 있었다.
[최종 코드]
[GitHub]
GitHub - SmallPeanutPark/BAEKJOON: Study BaekJoon for Coding Test
Study BaekJoon for Coding Test. Contribute to SmallPeanutPark/BAEKJOON development by creating an account on GitHub.
github.com
#include <bits/stdc++.h>
#define MODULAR 1000000
using namespace std;
int dp[5001]; // 길이에 따른 암호 코드 수, 인덱스가 길이를 의미
int main(void) {
ios_base::sync_with_stdio(false);
cin.tie(NULL);
string str = "0";
string inputstr;
cin >> inputstr;
str += inputstr;
int len = str.length();
if(str[1] == '0') {
// 맨 첫번째가 0일 때는 암호코드를 만들 수 없음 -> 0 출력
cout << 0;
return 0;
}
dp[0] = 1;
for (int i = 1; i < len; i++) {
if(i == 1) {
int cur_num = str[i] - '0';
if(cur_num >= 1 && cur_num <= 9) {
dp[i] = dp[i - 1] % MODULAR;
}
} else {
int cur_num = str[i] - '0';
int prev_num = str[i - 1] - '0';
int total_num = prev_num * 10 + cur_num;
if(cur_num == 0) {
if(prev_num == 1 || prev_num == 2) {
dp[i] = dp[i - 2] % MODULAR;
} else break;
} else {
if(total_num >= 11 && total_num <= 26) {
dp[i] = (dp[i - 1] + dp[i - 2]) % MODULAR;
} else {
dp[i] = dp[i - 1] % MODULAR;
}
}
}
}
cout << dp[len - 1];
return 0;
}
'PS > BaekJoon' 카테고리의 다른 글
[BaekJoon 1197번] 최소 스패닝 트리(C++) (0) | 2022.09.05 |
---|---|
[BaekJoon 1920번] 수 찾기(C++) (0) | 2022.08.31 |
[BaekJoon 1629번] 곱셈(C++) (0) | 2022.08.29 |
[BaekJoon 18352번] 특정 거리의 도시 찾기(C++) (0) | 2022.08.29 |
[BaekJoon 1717번] 집합의 표현(C++) (0) | 2022.08.28 |
댓글