POJ 3373 Changing Digits

#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>
#include <queue>
#include <cmath>
using namespace std;
const int maxn = 103, maxk = 10010;
int k, len, n[maxn], m[maxn];
int mod[maxn][10], flag[maxn][maxk];
char num[maxn];
void init_mod() {  //打表，用来计算余数
int i, j;
for(i = 0; i < 10; i++) {
mod[0][i] = i % k;
}
for(i = 1; i < len; i++) {
for(j = 0; j < 10; j++) {
mod[i][j] = mod[i - 1][j] * 10 % k;
}
}
}
int dfs(int pos, int restnum, int m_modk) {
if(m_modk == 0) return 1;  //找到满足的方案
if(restnum == 0 || pos < 0) return 0;  //已不满足条件
if(restnum <= flag[pos][m_modk]) return 0; //剪枝
int i, j;
for(i = pos; i >= 0; i--) {  //搜索比n小的数，要尽可能小，则从高位开始
for(j = 0; j < n[i]; j++) {
if(i == len - 1 && j == 0) continue;  //首位不能为0
int res = (m_modk - (mod[i][n[i]] - mod[i][j]) + k) % k;  //计算改变数字之后的m的余数
m[i] = j;
if(dfs(i - 1, restnum - 1, res)) return 1;
m[i] = n[i];
}
}
for(i = 0; i <= pos; i++) {
for(j = n[i] + 1; j < 10; j++) {  //搜索比n大的数，要尽可能小，则从低位开始
int res = (m_modk + (mod[i][j] - mod[i][n[i]]) + k) % k;  //注意这里也要加上k，为何...
m[i] = j;
if(dfs(i - 1, restnum - 1, res)) return 1;
m[i] = n[i];
}
}
flag[pos][m_modk] = restnum;  //能运行到这里说明搜索失败，更新剪枝数值
return 0;
}
int main() {
while(~scanf("%s%d", num, &k)) {
int i, j;
len = strlen(num);
init_mod();  //因为k不同，所以每次都要打表
int n_modk = 0;
for(i = 0; i < len; i++) {  //将n变成整形倒序存入数组n
n[i] = num[len - i - 1] - '0';
m[i] = n[i];  //同时记录m
n_modk = (n_modk + mod[i][n[i]]) % k; //计算n%k的余数
}
memset(flag, 0, sizeof(flag));  //初始化flag
for(i = 1; i <= len; i++) {  //从小到大枚举修改的位数，第一个满足的即为所求
if(dfs(len - 1, i, n_modk)) break;
}
for(i = len - 1; i >= 0; i--) {
printf("%d", m[i]);
}
printf("\n");
}
return 0;
}