bzoj3679 数字之积(数位dp+Hash表)

其实有效状态并不多,于是我们拿一个hash表记一下,直接数位dp即可。

#include <bits/stdc++.h>
using namespace std;
#define ll long long
#define inf 0x3f3f3f3f
#define N 10003
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;
}
int n,a[19];
ll L,R;
struct Hash_table{
    int next,key;ll val;
};
struct Icefox{
    int h[N],num;
    Hash_table data[N];
    inline void init(){memset(h,0,sizeof(h));num=0;}
    inline void ins(int key,ll val){
        int x=key%N;
        for(int i=h[x];i;i=data[i].next)
            if(data[i].key==key){data[i].val+=val;return;}
        data[++num].next=h[x];h[x]=num;data[num].key=key;data[num].val=val;
    }
}dp[19][2];
inline ll calc(ll m){
    int tot=0;ll ans=0;
    while(m) a[++tot]=m%10,m/=10;
    for(int i=1,j=tot;i<j;++i,--j) swap(a[i],a[j]);
    dp[0][0].init();dp[0][0].ins(1,1);
    for(int i=0;i<tot;++i){
        dp[i+1][0].init();dp[i+1][1].init();dp[i+1][1].ins(1,1);
        for(int j=1;j<=dp[i][0].num;++j){
            int x=dp[i][0].data[j].key;ll val=dp[i][0].data[j].val;
            for(int k=1;k<=a[i+1];++k){
                if(k>n/x) break;
                dp[i+1][k<a[i+1]].ins(x*k,val);
            }
        }for(int j=1;j<=dp[i][1].num;++j){
            int x=dp[i][1].data[j].key;ll val=dp[i][1].data[j].val;
            for(int k=1;k<=9;++k){
                if(k>n/x) break;
                dp[i+1][1].ins(x*k,val);
            }
        }
    }for(int i=1;i<=dp[tot][1].num;++i) ans+=dp[tot][1].data[i].val;
    return ans;
}
int main(){
//  freopen("a.in","r",stdin);
    n=read();L=read();R=read();
    printf("%lld\n",calc(R)-calc(L));
    return 0;
}

猜你喜欢

转载自blog.csdn.net/icefox_zhx/article/details/80715157