某区间内的数的各位之和取模后为0的数量（数位dp）

题目：https://www.acwing.com/problem/content/1086/
题意：某人又命名了一种取模数，这种数字必须满足各位数字之和 mod N 为 0。现在大家又要玩游戏了，指定一个整数闭区间 [a.b]，问这个区间内有多少个取模数。1≤a,b≤2^31 - 1 , 1≤N<100。

题解：数位dp。
首先，数位dp常规做法，只不过求左子树的时候，不太好求，求左子树的时候需要预处理一个dp数组，就是f[i][j][k]，表示总共有i位，最高位是j，取完模后是k的所有合法的数的数量，有了这个东西以后，直接暴力枚举这一位该选什么，去叠加对应的f数组即可。
f数组简单推导：
f[i][j][k]的最高位是j，假设这个数的次高位是x，那么这个x就是i-1位的最高位，同样那个取模那里也做个转换就是mod(k-j,P)，就是总的和(i位)减去前面那个最高位j，剩余的数再去取P，这样就可以f[i][j][k] += f[i - 1][x][mod(k-j,P)]了，mod是自定义一下取模函数，因为有负数情况，P是模数。这样枚举位数，枚举最高位，枚举取完模剩余的数，再枚举一下x就可以得到答案，四个for循环跑一下。

#include <bits/stdc++.h>
//#define int long long
#define pb push_back
#define pii pair<int, int>
#define mpr make_pair
#define ms(a, b) memset((a), (b), sizeof(a))
#define x first
#define y second
typedef long long ll;
typedef unsigned long long LL;
const int inf = 0x3f3f3f3f;
const ll INF = 0x3f3f3f3f3f3f3f3f;
using namespace std;

const int N = 11, M = 110;
// f[i j k]总共有i位，最高位为j，且各位数之和mod以后为k的方案数
int f[N][10][M];
int P;  //取模的数
//取模完可能为负数，自定义一个取模
int mod(int x, int y) { return (x % y + y) % y; }
void init() {
    memset(f, 0, sizeof(f));//多组测试数据
    //枚举1位的情况
    for (int i = 0; i <= 9; i++) f[1][i][i % P]++;
    //dp处理其他位
    for (int i = 2; i < N; i++) //枚举总共位数
        for (int j = 0; j <= 9; j++)  //枚举最高位是谁
            for (int k = 0; k < P; k++)  //枚举mod完以后的数
                for (int x = 0; x <= 9; x++)  // 最高位是j，次高位是x
                                              //同时也是i-1位的最高位的值
                    f[i][j][k] += f[i - 1][x][mod(k - j, P)];
}
int dp(int n) {
    if (!n) return 1;
    vector<int> nums;
    while (n) nums.emplace_back(n % 10), n /= 10;
    int res = 0;
    int last = 0;  //前面位数的和
    for (int i = nums.size() - 1; i >= 0; i--) {
        int x = nums[i];
        //左子树枚举这一位能够选的值，并且叠加f数组
        for (int j = 0; j < x; j++) res += f[i + 1][j][mod(-last, P)];
        //往右子树走，叠加x
        last += x; 
        //判断最后一个右节点，也就是判断n是不是满足条件
        if (!i && last % P == 0) res++;
    }
    //返回答案
    return res;
}
signed main() {
    int l, r;
    while (cin >> l >> r >> P) {
        init();
        printf("%d\n", dp(r) - dp(l - 1));
    }
    return 0;
}