bzoj1799 [Ahoi2009]self 同类分布（数位dp）

ahoi2009全ac达成嘿嘿qaq

因为我们要统计%M=0的数的个数，我们只好考虑枚举各位数字之和M，还好最大只有19*9，可以接受。
然后我们就可以数位dp了，f[0][i][j][k]表示前i位,数字和为j,%mod=k,目前还没< x的个数,f[1],已经< x了的个数。

复杂度$O(171^3*19)$

#include <cstdio>
#include <cstring>
#include <algorithm>
using namespace std;
#define N 1010
#define ll long long
#define inf 0x3f3f3f3f
inline char gc(){
    static char buf[1<<16],*S,*T;
    if(S==T){T=(S=buf)+fread(buf,1,1<<16,stdin);if(T==S) return EOF;}
    return *S++;
}
inline ll read(){
    ll x=0,f=1;char ch=gc();
    while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=gc();}
    while(ch>='0'&&ch<='9') x=x*10+ch-'0',ch=gc();
    return x*f;
}
ll f[2][20][200][200];//f[0][i][j][k],前i位,数字和为j,%mod=k,目前还没<x的个数
int a[N],tot=0;
inline ll calc(ll x){
    tot=0;while(x) a[++tot]=x%10,x/=10;ll res=0;
    for(int i=1,j=tot;i<j;++i,--j) swap(a[i],a[j]);
    for(int mod=1;mod<=9*tot;++mod){
        memset(f,0,sizeof(f));f[0][0][0][0]=1;
        for(int i=0;i<tot;++i)
            for(int j=0;j<=9*i;++j){
                if(j>mod) break;
                for(int k=0;k<mod;++k){
                    if(f[0][i][j][k]){
                        for(int p=0;p<=a[i+1];++p)
                            f[p<a[i+1]][i+1][j+p][(k*10+p)%mod]+=f[0][i][j][k];
                    }if(f[1][i][j][k]){
                        for(int p=0;p<=9;++p)
                            f[1][i+1][j+p][(k*10+p)%mod]+=f[1][i][j][k];
                    }
                }
            }res+=f[0][tot][mod][0]+f[1][tot][mod][0];
    }return res;
}
int main(){
//  freopen("a.in","r",stdin);
    ll x=read(),y=read();
    printf("%lld\n",calc(y)-calc(x-1));
    return 0;
}