hdu_5085_Counting problem(莫队分块思想)

题目连接：hdu_5085_Counting problem

题意：给你一个计算公式，然后给你一个区间，问这个区间内满足条件的数有多少个

题解：由于这个公式比较特殊，具有可加性，我们考虑讲一个数分为两个部分，这样就可以用莫队的思想均摊时间复杂度，将9位数分为一个4位和一个5位，这里我感觉sqr为10000 速度比较快。然后如果b小于sqr，那么直接暴力就行，如果b大于sqr,那么我们要把a和b都分为头部和尾部（注意是闭区间，a需要减1），如果a小于sqr，那么他的头部就为0，然后计算0-a的尾部，并将相应的值插入Hash表，然后计算以a头部开头满足条件的数，记为reta，同理，计算0-b的尾部，记为retb，如果b尾小于a尾，就要先计算b再计算a，然后将0到sqr-1的数对应的值也全部插入Hash表，然后从a的头到b的头，寻找对应的值，这里要用一下容斥定理，retb-reta就是从[a,b]之间满条件的数的个数.

#include<cstdio>
#define F(i,a,b) for(int i=a;i<=b;i++)
typedef long long LL;

const LL M=(1<<20)-1,N=1e5+7,sqr=1e4;
LL K,a,b,ahd,bhd,bt,at,reta,retb,T,S,dt[10][16];
inline LL pow(LL x,LL k){
    LL an=1;
    while(k){
        if(k&1)an*=x;
        k>>=1,x*=x;
    }
    return an;
}
//------------Hash table------------------------
struct E{LL key;LL cnt;E *nxt;}*g[M+1],pool[N],*cur=pool,*p;LL vis[M+1];
void init_Hash(){T++,cur=pool;}
inline void ins(LL key){
    if(key>S)return;
    LL u=key&M;
    if(vis[u]<T)vis[u]=T,g[u]=NULL;
    for(p=g[u];p;p=p->nxt)if(p->key==key){p->cnt++;return;}
    p=cur++,p->key=key,p->cnt=1,p->nxt=g[u],g[u]=p;
}

inline LL ask(LL key){
    if(key<0)return 0;
    LL u=key&M;
    if(vis[u]<T)return 0;
    for(p=g[u];p;p=p->nxt)if(p->key==key)return p->cnt;
    return 0;
}
//----------------------------------------------
inline LL cal(LL x){
    LL an=0;
    while(x)an+=dt[x%10][K],x/=10;
    return an;
}
void init(){F(i,0,9)F(j,1,15)dt[i][j]=pow(i,j);}
int main(){
    init();
    while(~scanf("%lld%lld%lld%lld",&a,&b,&K,&S)){
        init_Hash();
        ahd=(a-1)/sqr,bhd=b/sqr,at=(a-1)%sqr,bt=b%sqr;
        if(at<bt){
        F(i,0,at)ins(cal(i));
        reta=ask(S-cal(ahd));
        F(i,at+1,bt)ins(cal(i));
        retb=ask(S-cal(bhd));
        F(i,bt+1,sqr-1)ins(cal(i));
    }else{
        F(i,0,bt)ins(cal(i));
        retb=ask(S-cal(bhd));
        F(i,bt+1,at)ins(cal(i));
        reta=ask(S-cal(ahd));
        F(i,at+1,sqr-1)ins(cal(i));
    }
    F(i,ahd,bhd-1)retb+=ask(S-cal(i));
    printf("%lld\n",retb-reta);
    }
    return 0;
}