bzoj1799: [Ahoi2009]self 同类分布 数位Dp

bzoj1799: [Ahoi2009]self 同类分布

Description

给出a,b，求出[a,b]中各位数字之和能整除原数的数的个数。

Sample Input

10 19

Sample Output

3

HINT

【约束条件】1 ≤ a ≤ b ≤ 10^18

分析

数字和个数是有限的。枚举一下。
转化为两个数之间各个数位之和和数字本身是某个数的倍数的数的个数。上数位Dp
$F[i][j][k]$表示第i位，当前各个数字位数之和以及当前数的大小。
注意到我们并不一定要知道数的大小。我们只需要知道这个数是否为当前枚举的数的倍数。
于是可以在mod意义下转移，然后就搞定了。
复杂度很妥。

代码

#include<cstdio>
#include<algorithm>
#include<cstdlib>
long long f[20][220][220];
int v[20][220][220], bit[20], T, M;
long long Dp(int i, int s, int d, bool e) {
    if(!i) return !d && !s;
    if(!e && v[i][s][d] == T) return f[i][s][d];
    int l = std::max(0, s - (i - 1) * 9), r = std::min(e ? bit[i] : 9, s); long long R = 0;
    for(int j = l;j <= r; ++j) R += Dp(i - 1, s - j, (d * 10 + j) % M, e & (bit[i] == j));
    if(!e) v[i][s][d] = T, f[i][s][d] = R; return R;
}
long long Cal(long long x) {
    int tp = 0; for(;x; x /= 10) bit[++tp] = x % 10;
    long long r = 0; for(M = 1;M <= tp * 9; ++M) ++T, r += Dp(tp, M, 0, 1);
    return r;
}
int main() {
    long long A, B; scanf("%lld%lld", &A, &B);
    printf("%lld\n", Cal(B) - Cal(A - 1));
    return 0;
}