bzoj 1799: [Ahoi2009]self 同类分布【数位DP*好题】

1799: [Ahoi2009]self 同类分布
Time Limit: 50 Sec Memory Limit: 64 MB
Submit: 2152 Solved: 957
[Submit][Status][Discuss]
Description
给出a,b，求出[a,b]中各位数字之和能整除原数的数的个数。
Input
Output
Sample Input
10 19
Sample Output
3
【约束条件】1 ≤ a ≤ b ≤ 10^18

分析：枚举和我想到了，但是DP让我写挂了，菜~~~
$dp[pos][x][val]：前pos位余数为x，和为val能整除的个数$。所以，暴力到pos位，余数为x(那么前pos位就是k-x)，和为val(用的减法)就能剪枝了。
我之前写的是dp[pos][k][val]，把枚举的和当成一维了，仔细想，这样是很明显不对的。和的那一维可以省掉的，那么就剩下dp[pos][val]，就只考虑数位和了，没有考虑到取余这个操作。

#pragma GCC optimize ("O3")
#pragma GCC optimize ("O2")
#include <bits/stdc++.h>
#include <ext/rope>
using namespace std;
using namespace __gnu_cxx;
#define met(s) memset(s, 0, sizeof(s))
#define rep(i, a, b) for(int i = a; i <= b; ++i)
template <class T> inline void scan_d(T &ret) {
char c; ret = 0;
while ((c = getchar()) < '0' || c > '9');
while (c >= '0' && c <= '9') {
ret = ret * 10 + (c - '0'), c = getchar();}}
typedef long long LL;
typedef unsigned long long ull;
typedef pair<int, int> pii;
const int INF = 0x3f3f3f3f;
const int mod = 1e9 + 7;
const int MAXN = 1e4 + 10;
LL dp[20][200][200];
int a[20];

inline LL dfs(int pos, int k, int x, int val, int limit) {
    //if(val - (pos + 1) * 9 > 0) return 0;
    if(pos < 0) return !val && !x;
    if(!limit && dp[pos][k - x][val] >= 0) return dp[pos][k - x][val];
    int up = limit ? a[pos] : 9;
    LL ans = 0;
    for(int i = 0; i <= up; ++i) {
        if(val - i < 0) break;
        ans += dfs(pos - 1, k, (x * 10 + i) % k, val - i, limit && i == up) * 1LL;
    }
    if(!limit) dp[pos][k - x][val] = ans;
    return ans;
}

inline LL so(LL x) {
    int p = 0;
    while(x) {
        a[p++] = x % 10;
        x /= 10;
    }
    LL ans = 0;
    for(int i = 1; i <= p * 9; ++i) {
        memset(dp, -1, sizeof(dp));
        ans += dfs(p - 1, i, 0, i, 1);
    }
    return ans;
}

int main() {
    LL a, b;
    scanf("%lld %lld", &a, &b);
    printf("%lld\n", so(b) - so(a - 1));
    return 0;
}