poj 3373 Changing Digits(记忆化搜索)

题目链接：http://poj.org/problem?id=3373

参考了网上的一篇博客，感觉讲得很详细

附上传送门：http://blog.csdn.net/lyy289065406/article/details/6698787/

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<queue>
using namespace std;

const int INF=0x3f3f3f3f;
const int maxn=105;
char s[maxn];
int k,len;
int dp[maxn][15];
int flag[maxn][10010];//flag[i][j]=cnt表示从i开始取cnt个数不满足j=0
int a[maxn],ans[maxn];

void init(){
	memset(flag,0,sizeof(flag));
	for(int j=0;j<=9;j++)
		dp[0][j]=j%k;
	for(int i=1;i<len;i++)
		for(int j=0;j<=9;j++)
			dp[i][j]=(dp[i-1][j]*10)%k;
	return ;
}

bool dfs(int cnt,int l,int mod){
	if(!mod){
		for(int i=0;i<len;i++)
			printf("%d",ans[i]);
		putchar(10);
		return true;
	}
	if(!cnt) return false;
	if(l>len-1) return false;
	if(flag[l][mod]>=cnt) return false;//剪枝，如果从l开始取cnt个数不满足mod==0，则小于cnt个数必然不满足
	for(int i=l;i<len;i++){//从高位开始取使其变小
		for(int j=0;j<a[i];j++){
			if(!i&&!j) continue;
			ans[i]=j;
			int tmp=(mod-(dp[len-i-1][a[i]]-dp[len-i-1][j])+k)%k;
			if(dfs(cnt-1,i+1,tmp))
				return true;
		}
		ans[i]=a[i];
	}
	for(int i=len-1;i>=l;i--){//若无法变小，则使其变大，从低位开始取
		for(int j=a[i]+1;j<=9;j++){
			if(!i&&!j) continue;
			ans[i]=j;
			int tmp=(mod-(dp[len-i-1][a[i]]-dp[len-i-1][j])+k)%k;
			if(dfs(cnt-1,i+1,tmp))
				return true;
		}
		ans[i]=a[i];
	}
	flag[l][mod]=cnt;
	return false;
}

int main(){
#ifndef ONLINE_JUDGE
	freopen("test.in","r",stdin);
	freopen("test.out","w",stdout);
#endif
	while(~scanf("%s%d",s,&k)){
		len=strlen(s);
		init();
		for(int i=0;i<len;i++)
			ans[i]=a[i]=s[i]-'0';
		int mod=0;
		for(int i=len-1;i>=0;i--)
			mod=(mod+dp[len-i-1][a[i]])%k;
		for(int i=0;i<len;i++)
			if(dfs(i,0,mod))
				break;
	}
	return 0;
}