ZOJ-3962-数位dp

http://acm.zju.edu.cn/onlinejudge/showProblem.do?problemId=5594

 

 

 

 

 

 

　　　　16进制下的数位dp，由于固定了位数，可以出现前导零，反而简化了问题，和十进制异曲同工，只需要注意边界，对于[l,r],如果r>max,则分成[l,max-1]+max+[0,r-max-1],这是因为对于cal(N,x)函数来说计算的是[0,N)之间x的出现次数,并不包括N，如果只是简单的把max+1传进去的话就不是八位数了，这

样乱改函数的话更麻烦还不如直接把max抽出来计算方便。

　　　　

 #include<bits/stdc++.h>
 using namespace std;
 #define LL long long
 LL c[]={,,,,,,,,
           ,,,,,,,};
 LL f[]={,};
 LL p16[]={};
 LL bit[];
 void init(){
     for(LL i=;i<=;++i) p16[i]=p16[i-]*;
     for(LL i=;i<=;++i) f[i]=f[i-]*+p16[i-];
 }
 LL cal(LL N,LL x){
     LL len=,ans=,tot=;
     while(N){
         bit[len++]=N%;
         N/=;
     }
     while(len<) bit[len++]=;
     bit[len]=-;
     for(LL i=len-;i>=;--i){
         ans+=f[i]*bit[i];
         if(bit[i]>x) ans+=p16[i];
         ans+=tot*bit[i]*p16[i];
         if(bit[i]==x) tot++;
     }
     return ans;
 }
 LL to10(char *s){
     LL ans=,len=strlen(s);
     for(int i=;i<len;++i){
         LL tmp=isdigit(s[i])?s[i]-'':(s[i]-'A'+);
         ans=ans*+tmp;
     }
     return ans;
 }
 int main(){init();
     char s[];
     LL l,r,t,n,i,j,k;
     LL MAX=to10("FFFFFFFF");
     scanf("%lld",&t);
     while(t--){
         scanf("%lld",&k);
         scanf("%s",s);
         LL l=to10(s);
         LL r=l+k-;
         LL ans=;

         if(r<=MAX){
             for(i=;i<=;++i) ans+=c[i]*(cal(r+,i)-cal(l,i));
         }
         else{
             for(i=;i<=;++i) if(MAX>=l)ans+=c[i]*(cal(MAX,i)-cal(l,i));
             ans+=c[]*;
             for(i=;i<=;++i) ans+=c[i]*cal(r-MAX,i);
         }
         printf("%lld\n",ans);
     }
     return ;
 }
 /*
 10
 5 89ABCDEF
 3 FFFFFFFF
 7 00000000
 */