题目连接: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]之间满条件的数的个数.
![](https://i-blog.csdnimg.cn/blog_migrate/8f900a89c6347c561fdf2122f13be562.gif)
![](https://i-blog.csdnimg.cn/blog_migrate/961ddebeb323a10fe0623af514929fc1.gif)
1 #include<cstdio> 2 #define F(i,a,b) for(int i=a;i<=b;i++) 3 typedef long long LL; 4 5 const LL M=(1<<20)-1,N=1e5+7,sqr=1e4; 6 LL K,a,b,ahd,bhd,bt,at,reta,retb,T,S,dt[10][16]; 7 inline LL pow(LL x,LL k){ 8 LL an=1; 9 while(k){ 10 if(k&1)an*=x; 11 k>>=1,x*=x; 12 } 13 return an; 14 } 15 //------------Hash table------------------------ 16 struct E{LL key;LL cnt;E *nxt;}*g[M+1],pool[N],*cur=pool,*p;LL vis[M+1]; 17 void init_Hash(){T++,cur=pool;} 18 inline void ins(LL key){ 19 if(key>S)return; 20 LL u=key&M; 21 if(vis[u]<T)vis[u]=T,g[u]=NULL; 22 for(p=g[u];p;p=p->nxt)if(p->key==key){p->cnt++;return;} 23 p=cur++,p->key=key,p->cnt=1,p->nxt=g[u],g[u]=p; 24 } 25 26 inline LL ask(LL key){ 27 if(key<0)return 0; 28 LL u=key&M; 29 if(vis[u]<T)return 0; 30 for(p=g[u];p;p=p->nxt)if(p->key==key)return p->cnt; 31 return 0; 32 } 33 //---------------------------------------------- 34 inline LL cal(LL x){ 35 LL an=0; 36 while(x)an+=dt[x%10][K],x/=10; 37 return an; 38 } 39 void init(){F(i,0,9)F(j,1,15)dt[i][j]=pow(i,j);} 40 int main(){ 41 init(); 42 while(~scanf("%lld%lld%lld%lld",&a,&b,&K,&S)){ 43 init_Hash(); 44 ahd=(a-1)/sqr,bhd=b/sqr,at=(a-1)%sqr,bt=b%sqr; 45 if(at<bt){ 46 F(i,0,at)ins(cal(i)); 47 reta=ask(S-cal(ahd)); 48 F(i,at+1,bt)ins(cal(i)); 49 retb=ask(S-cal(bhd)); 50 F(i,bt+1,sqr-1)ins(cal(i)); 51 }else{ 52 F(i,0,bt)ins(cal(i)); 53 retb=ask(S-cal(bhd)); 54 F(i,bt+1,at)ins(cal(i)); 55 reta=ask(S-cal(ahd)); 56 F(i,at+1,sqr-1)ins(cal(i)); 57 } 58 F(i,ahd,bhd-1)retb+=ask(S-cal(i)); 59 printf("%lld\n",retb-reta); 60 } 61 return 0; 62 }