hdu 3652 B-number （数位DP）

最新推荐文章于 2022-04-05 09:12:07 发布

原创最新推荐文章于 2022-04-05 09:12:07 发布 · 453 阅读

1 ·

CC 4.0 BY-SA版权

DP 专栏收录该内容

27 篇文章

订阅专栏

本文讨论了使用记忆化搜索技术解决HDOJ（HDU Online Judge）平台上的问题，具体涉及到输入数据处理、状态转移方程设计以及记忆化数组的应用。

题目链接：hdu 3652

我还是只会用记忆化搜索写= =。

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

int limit[100],dp[20][15][5];

int dfs(int l,int mod,int now,int fp){
	if(!l)return (!mod && now == 2);
	if(!fp && dp[l][mod][now]!=-1)return dp[l][mod][now];
	int tot=0;int lmax=fp?limit[l]:9;
	for(int i=0;i<=lmax;i++){
		int sai=now;
		if(now==1&&i!=1)sai=0;
		if(now==0&&i==1)sai=1;
		if(now==1&&i==3)sai=2;
		tot+=dfs(l-1,(mod*10+i)%13,sai,fp && i==lmax);
	}
	if(!fp)dp[l][mod][now]=tot;
	return tot;
}

int find(int x){
	int len=0;
	while(x){
		limit[++len]=x%10; x/=10;
	}
	return dfs(len,0,0,1);
}

int main(){
	int N;
	while(scanf("%d",&N)==1){
		memset(dp,-1,sizeof(dp));
		printf("%d\n",find(N));
	}
	return 0;
}