题意:给一个数字,问有多少个小于等于这个数且满足包含13这个子串,满足被13整除的数的个数。
数位DP,记忆化搜索实现,dp[i][j][k],i代表位置,j代表mod13剩下的值,k有3个值,0代表不包含13,1代表不包含13,但是当前数是3,2代表含有13了。。
#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
int dp[15][15][3];
int dig[15];
int dfs(int pos,int prenum,int premod,bool up,bool flag) //up达到了上界没有,flag是否包含了13
{
if(pos<=0)
return (flag&&premod==0);
if(!up&&flag&&dp[pos][premod][0]!=-1) //前面已经包含了13,所以后面的可以不包含
return dp[pos][premod][0];
if(!up&&!flag&&prenum==1&&dp[pos][premod][1]!=-1) //前一个是1,后面紧跟着添个3就好
return dp[pos][premod][1];
if(!up&&!flag&&prenum!=1&&dp[pos][premod][2]!=-1) //什么都没有,返回包含13的
return dp[pos][premod][2];
int end=up?dig[pos]:9;
int res=0;
for(int i=0;i<=end;i++)
res+=dfs(pos-1,i,(premod*10+i)%13,up&&(i==end),flag||(prenum==1&&i==3));
if(!up)
{
if(!flag&&prenum==1)
dp[pos][premod][1]=res;
if(!flag&&prenum!=1)
dp[pos][premod][2]=res;
if(flag)
dp[pos][premod][0]=res;
}
return res;
}
int cal(int num)
{
int k=0;
while(num)
{
dig[++k]=num%10;
num=num/10;
}
return dfs(k,0,0,1,0);
}
int main()
{
int n;
memset(dp,-1,sizeof(dp));
while(scanf("%d",&n)==1)
{
printf("%d\n",cal(n));
}
return 0;
}