hdu 3652 B-number

http://acm.hdu.edu.cn/showproblem.php?pid=3652

数位dp

题意：求1到n中能被13整除且含有13的数的个数。dp[i][j][k][c]表示dfs到i位，余数为j，是否含有13的标志k，最后一个数为m的有多少个符合要求的数。

dfs枚举每一位。

#include <cstdio>
#include <cstring>
#define ll int
#include <algorithm>
using namespace std;

ll n;
ll dp[15][15][10][15];
int num[30];

ll dfs(int pos,int c,bool x,int m,bool flag)
{
    if(pos==-1) return (x&&c==0);
    if(!flag&&dp[pos][c][x][m]!=-1) return dp[pos][c][x][m];
    ll ans=0;
    int xx=flag?num[pos]:9;
    for(int i=0; i<=xx; i++)
    {
        ans+=dfs(pos-1,(c*10+i)%13,x||(m==1&&i==3),i,flag&&(i==xx));
    }
    if(!flag) dp[pos][c][x][m]=ans;
    return ans;
}


int main()
{
    memset(dp,-1,sizeof(dp));
    while(scanf("%d",&n)!=EOF)
    {
        int cnt=0;
        while(n)
        {
            num[cnt++]=n%10;
            n=n/10;
        }
        printf("%d\n",dfs(cnt-1,0,false,0,true));
    }
    return 0;
}