Light OJ 1068

http://www.lightoj.com/volume_showproblem.php?problem=1068

类型:数位DP

#include <cstdio>
#include <algorithm>
#include <cstring>
using namespace std;
#define ll unsigned int
ll dp[83][11][83][83],md[12]={1};
void init()
{
    memset(dp,0,sizeof(dp));
    for(int i=1;i<11;i++)
    md[i]=10*md[i-1];
    for(int k=1;k<83;k++)
        for(int i=0;i<10;i++)
            dp[k][1][i%k][i%k]++;
    for(int k=2;k<83;k++)
    for(int i=2;i<11;i++)
    for(int j=0;j<k;j++)
    for(int l=0;l<k;l++)
    for(int m=0;m<10;m++){
        int tmp1=(m*md[i-1])%k;
        int tmp2=m%k;
        dp[k][i][j][l]+=dp[k][i-1][(k+j-tmp2)%k][(k+l-tmp1)%k];
    }  
}
void print()
{
    for(int k=2;k<5;k++)
    for(int i=1;i<3;i++)
    for(int j=0;j<k;j++)
    for(int l=0;l<k;l++)
    printf("k=%d i=%d j=%d l=%d ans=%d
",k,i,j,l,dp[k][i][j][l]);
}
ll solve(ll x,ll k)
{
    ll dig[15],cnt=0,ans=0;
    while(x){
        dig[++cnt]=x%10;
        x/=10;
    }
    ll sum=0,mul=0;
    for(int i=cnt;i>1;i--){
        for(int j=0;j<dig[i];j++){
            ll tmp1=(sum+j)%k;
            ll tmp2=(mul+j*md[i-1])%k;
            ans+=dp[k][i-1][(k-tmp1)%k][(k-tmp2)%k];
        }
       
        sum=(sum+dig[i])%k;
        mul=(mul+dig[i]*md[i-1])%k;
    }
    for(int i=0;i<dig[1];i++)
        if((mul+i)%k==0&&(sum+i)%k==0)
            ans++;
    return ans;
}
int main()
{
    init();
    int t,cas=1;
    scanf("%d",&t);
    while(t--){
        ll l,r,k;
        scanf("%d%d%d",&l,&r,&k);
        printf("Case %d: ",cas++);
        if(k==1)
        printf("%d
",r-l+1);
        else if(k>82)
        printf("0
");
        else
        printf("%u
",solve(r+1,k)-solve(l,k));
    }
    return 0;
}