POJ 3373 Changing Digits

POJ 3373


参考博客:http://blog.csdn.net/lyy289065406/article/details/6698787     解释得非常详细,大部分都能看懂。 感谢!

  http://www.cnblogs.com/fenshen371/p/3250109.html   感觉代码写的很漂亮。  感谢!

感觉不太好做,加油吧。

#include 
#include 
#include 
#include 
#include 
#include 
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;
}


你可能感兴趣的:(深搜,搜索,剪枝)