1084. 数字游戏 II(数位dp)

由于科协里最近真的很流行数字游戏。

某人又命名了一种取模数，这种数字必须满足各位数字之和 mod N 为 0。

现在大家又要玩游戏了，指定一个整数闭区间 [a.b]，问这个区间内有多少个取模数。

输入格式
输入包含多组测试数据，每组数据占一行。

每组数据包含三个整数 a,b,N。

输出格式
对于每个测试数据输出一行结果，表示区间内各位数字和 mod N 为 0 的数的个数。

数据范围
1≤a,b≤231−1,
1≤N<100

输入样例：
1 19 9
输出样例：
2

#include<bits/stdc++.h>
using namespace std;
const int N = 11;
const int MOD = 100;
int f[N][10][MOD];
int mo,l,r;
int mod(int x,int y){
    return (x % y + y) % y;
}
void init(){
    memset(f,0,sizeof f);
    for(int i = 0;i < 10;i ++){
        for(int j = 0;j < MOD;j ++){
            if(mod(i,mo) == j)f[1][i][j] ++;
        }
    }
    for(int i = 2;i < N;i ++){
        for(int j = 0;j < 10;j ++){
            for(int k = 0;k < MOD;k ++){
                for(int z = 0;z < 10;z ++){
                    f[i][j][k] += (f[i - 1][z][mod(k - j,mo)]);
                }
            }
        }
    }
}
int dp(int n){
    if(!n)return 1;
    vector<int>nums;
    while(n)nums.push_back(n % 10),n /= 10;
    
    int res = 0,last = 0;
    for(int i = nums.size() - 1;i >= 0;i --){
        int x = nums[i];
        for(int j = 0;j < x;j ++)res += f[i + 1][j][mod(last,mo)];
        last = mod(last - x,mo);
        if(!i && !last)res ++;
    }
    
    return res;
}
int main(){
    while(cin>>l>>r>>mo){
        if(l > r)swap(l,r);
        init();
        cout<<(dp(r) - dp(l - 1))<<endl;
    }
    return 0;
}