【BZOJ】1799: [Ahoi2009]self 同类分布

【题意】给出a,b，求出[a,b]中各位数字之和能整除原数的数的个数。1 ≤ a ≤ b ≤ 10^18

【算法】数位DP

【题解】

感觉这种方法很暴力啊。

枚举数位和1~162（不能枚举0，不然会模0，相当于除0），记忆化f[pos][sum][val]，sum表示当前数位和，val表示数字取模枚举的数位和。

每次sum+i和(val*10+i)%MOD转移。

sum用减法优化，即记忆化(MOD-sum)，但是枚举过程中都要memset，导致效率低下，记忆化效果很差。

要什么方法才能跑1.3s啊，%%%。

#include<cstdio>
#include<algorithm>
#include<cstring>
#define ll long long
using namespace std;
const int maxn=;
ll f[maxn][][],a[maxn],b[maxn],c[maxn],n,MOD;
ll dfs(int pos,int sum,int val,int limit){
    if(sum>MOD)return ;
    if(pos==-){if(sum==MOD&&val==)return ;else return ;}
    if(!limit&&~f[pos][MOD-sum][val])return f[pos][MOD-sum][val];

    int up=limit?a[pos]:;
    ll ans=;
    for(int i=;i<=up;i++){
        ans+=dfs(pos-,sum+i,(val*+i)%MOD,limit&&i==up);
    }
    if(!limit)f[pos][MOD-sum][val]=ans;
    return ans;
}
int main(){
    ll A,B,cntb=,cntc=;
    scanf("%lld%lld",&A,&B);
    A--;
    while(A){
        b[cntb++]=A%;
        A/=;
    }
    while(B){
        c[cntc++]=B%;
        B/=;
    }
    ll ans=;
    for(int i=;i<=;i++){//不能模0啊！！！模0也是除0啊！！！
        memset(f,-,sizeof(f));
        MOD=i;
        n=cntb;
        for(int j=;j<n;j++)a[j]=b[j];
        ans-=dfs(n-,,,);
        n=cntc;
        for(int j=;j<n;j++)a[j]=c[j];
        ans+=dfs(n-,,,);
    }
    printf("%lld",ans);
    return ;
}