HDOJ 3652 B-number (数位DP)

#include <stdio.h>
#include <string.h>
#define N 10
int dp
[13][3],digit
;
int dfs(int pos,int r,int s,int f)
{
if(pos==-1) return r==0&&s==2;
if(!f&&dp[pos][r][s]!=-1)   return dp[pos][r][s];
int max=f?digit[pos]:9;
int ret=0;
for(int i=0;i<=max;i++)
{
int nr=(r*10+i)%13;
int ns=s;
if(s==0&&i==1)  ns=1;
else if(s==1&&i!=1) ns=0;
if(s==1&&i==3)  ns=2;
ret+=dfs(pos-1,nr,ns,f&&i==max);
}
if(!f)  dp[pos][r][s]=ret;
return ret;
}
int cal(int x)
{
int pos=0;
while(x)
{
digit[pos++]=x%10;
x/=10;
}
return dfs(pos-1,0,0,1);
}
int main()
{
int x;
memset(dp,-1,sizeof(dp));
while(~scanf("%d",&x))
{
printf("%d\n",cal(x));
}
return 0;
}