bzoj1799[Ahoi2009]self 同类分布 【数位dp】

链接：http://www.lydsy.com/JudgeOnline/problem.php?id=1799

分析：设状态dp[i][j][k]，表示长度为i，所有位数和为j，当前模的余数为k，因为所有位的和最大为18*9，所以枚举模数，用记忆化搜索找到最后k==0的数。

代码：

#include<algorithm>
#include<iostream>
#include<cstring>
#include<cstdio>
#include<string>
#include<vector>
#include<queue>
#include<cmath>
#include<stack>
#include<set>
#include<map>
#define INF 0x3f3f3f3f
#define Mn 21
#define Mm 163
#define CLR(a,b) memset((a),(b),sizeof((a)))
#define CLRS(a,b,Size) memset((a),(b),sizeof((a[0]))*(Size+1))
#define CPY(a,b) memcpy ((a), (b), sizeof((a)))
#pragma comment(linker, "/STACK:102400000,102400000")
#define ul u<<1
#define ur (u<<1)|1
using namespace std;
typedef long long ll;
int a[Mn];
ll dp[Mn][Mm][Mm];
int vis[Mn][Mm][Mm];
ll cal(int len,int sum,int k,int isM,int mod) {
    if(!len) return k==0&&sum==mod;
    if(!isM) {
        if(vis[len][sum][k]) return dp[len][sum][k];
        vis[len][sum][k]=1;
        for(int i=0;i<=9;i++) if(sum+i>mod) break;
             else dp[len][sum][k]+=cal(len-1,sum+i,(k*10+i)%mod,0,mod);
        return dp[len][sum][k];
    } else {
        ll re=0;
        for(int i=0;i<=a[len];i++)if(sum+i>mod) break;
             else re+=cal(len-1,sum+i,(k*10+i)%mod,i==a[len],mod);
        return re;
    }
}
ll solve(ll x) {
    if(x==0) return 0;
    int cnt=0;
    ll re=0;
    while(x) {
        a[++cnt]=x%10;
        x/=10;
    }
    for(int i=1;i<=cnt*9;i++) {
        CLR(dp,0);
        CLR(vis,0);
        for(int j=0;j<=a[cnt];j++)
            re+=cal(cnt-1,j,j%i,j==a[cnt],i);
    }
    return re;
}
int main() {
    ll l,r;
    scanf("%lld%lld",&l,&r);
    printf("%lld",solve(r)-solve(l-1));
}