BZOJ 1799 同类分布

一开始没想出来。。一看题解

我艹直接枚举数位的和啊。。。。。怪不得给50s。

还是太蠢。

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
long long l,r,bit[],ret=,dp[][][],vis[][][],tot=;
void get_bit(long long x)
{
    ret=;
    while (x) {bit[++ret]=x%;x/=;}
}
long long dfs(long long pos,long long ret1,long long ret2,bool flag,long long mod)
{
    if (!pos) return ((ret1==mod) && (!ret2));
    if ((!flag) && (vis[pos][ret1][ret2]==tot)) return dp[pos][ret1][ret2];
    long long ans=,up=flag?bit[pos]:;
    for (long long i=;i<=up;i++)
        ans+=dfs(pos-,ret1+i,(ret2*+i)%mod,flag&&(i==bit[pos]),mod);
    if (!flag) dp[pos][ret1][ret2]=ans,vis[pos][ret1][ret2]=tot;
    return ans;
}
long long work(long long x)
{
    get_bit(x);
    long long ans=;
    for (long long i=;i<=ret*;i++)
    {
        tot++;
        ans+=dfs(ret,,,,i);
    }
    return ans;
}
int main()
{
    scanf("%lld%lld",&l,&r);
    printf("%lld\n",work(r)-work(l-));
    return ;
}