bzoj 1833 [ZJOI2010]count 数字计数

#include<iostream>
#include<cstdio>
#include<cstring>
#define ll long long
using namespace std;
const int N=12;
ll l,r,s[N+5],c[N+5],ct,sum[15],nm[N+5];
ll pw(int k)
{
    ll mul=10,ret=1;//ll
    while(k)
    {
        if(k&1)ret*=mul;
        mul*=mul;k>>=1;
    }
    return ret;
}
int cal(ll a)
{
    if(!a)return 1;
    int cnt=0;
    while(a)a/=10,cnt++;
    return cnt;
}
void init()
{
    int cnt=cal(r);
    for(int i=1;i<=cnt;i++)
        s[i]=s[i-1]*10+pw(i-1);
    for(int i=2;i<=cnt;i++)
        c[i]=pw(i-1)+c[i-1];
}
int pre(ll a)
{
    int cnt=cal(a);
    for(int i=1;i<=cnt;i++)nm[i]=a%10,a/=10;
    return cnt;
}
int main()
{
    scanf("%lld%lld",&l,&r);
    init();ct=pre(r);
    for(int i=ct;i;i--)
    {
        for(int j=0;j<=9;j++)
            sum[j]+=nm[i]*s[i-1]+(nm[i]>j)*pw(i-1);//nm[i]表示从0到nm[i]-1 
        sum[nm[i]]+=(r%pw(i-1))+1;
    }
    sum[0]-=c[ct];
    if(l)
    {
        ct=pre(l-1);
        for(int i=ct;i;i--)
        {
            for(int j=0;j<=9;j++)
                sum[j]-=nm[i]*s[i-1]+(nm[i]>j)*pw(i-1);//这里是+！(因为是-=) 
            sum[nm[i]]-=((l-1)%pw(i-1))+1;
        }
        sum[0]+=c[ct];
    }
    for(int i=0;i<=9;i++)printf("%lld ",sum[i]);
    return 0;
}