hdu 3652 B-number 数位dp

题目链接

求出1-n中包含13并且能被13整除的数的个数

开一个四维数组dp[i][j][k][l]， i表示第i位， j表示这个数mod13， k表示是否包含13， l表示前一位是什么。

#include<bits/stdc++.h>
using namespace std;
#define pb(x) push_back(x)
#define ll long long
#define mk(x, y) make_pair(x, y)
#define lson l, m, rt<<1
#define mem(a) memset(a, 0, sizeof(a))
#define rson m+1, r, rt<<1|1
#define mem1(a) memset(a, -1, sizeof(a))
#define mem2(a) memset(a, 0x3f, sizeof(a))
#define rep(i, a, n) for(int i = a; i<n; i++)
#define ull unsigned long long
typedef pair<int, int> pll;
const double PI = acos(-1.0);
const double eps = 1e-8;
const int mod = 1e9+7;
const int inf = 1061109567;
const int dir[][2] = { {-1, 0}, {1, 0}, {0, -1}, {0, 1} };
int dp[20][30][2][10], digit[20];
int dfs(int len, int num, int pre, bool ok, bool fp) {
    if(!len) {
        return ok&&num==0;
    }
    if(!fp&&dp[len][num][ok][pre]!=-1)
        return dp[len][num][ok][pre];
    int maxx = fp?digit[len]:9, ret = 0;
    for(int i = 0; i<=maxx; i++) {
        ret += dfs(len-1, (num*10+i)%13, i, ok||(pre==1&&i==3), fp&&i==maxx);
    }
    if(!fp)
        return dp[len][num][ok][pre] = ret;
    return ret;
}
int cal(int n) {
    int len = 0;
    while(n) {
        digit[++len] = n%10;
        n/=10;
    }
    return dfs(len, 0, 0, 0, 1);
}
int main()
{
    int a;
    mem1(dp);
    while(scanf("%d", &a)!=EOF) {
        printf("%d\n", cal(a));
    }
}