bzoj 1833: [ZJOI2010]count 数字计数

#include<cstdio>
#include<iostream>
#define ll long long
using namespace std;
struct data
{
    ll s[10];
}a[15][10],sum1,sum2;
ll f[18],l,r;
data operator +(data a1,data a2)
{
    data t;
    for(int i=0;i<=9;i++)
      t.s[i]=a1.s[i]+a2.s[i];
    return t;
}
data suan(ll r)
{
    int k;
    data t;
    for(int i=0;i<=9;i++)
      t.s[i]=0;
    if(!r)
      {
        t.s[0]=1;
        return t;
      }
    for(k=15;f[k]>r;k--);
    for(int i=1;i<k;i++)
      for(int j=1;j<=9;j++)
        t=t+a[i][j];
    t.s[0]++;
    ll s12=r/f[k];
    for(int i=1;i<s12;i++)
      t=t+a[k][i];
    ll s11=r%f[k];
    t.s[s12]+=s11+1;
    for(k--;k;k--)
      {
        s12=s11/f[k];
          for(int i=0;i<s12;i++)
            t=t+a[k][i];
        s11%=f[k];
        t.s[s12]+=s11+1;
      }
    return t;
}
int main()
{
    scanf("%lld%lld",&l,&r);
    f[1]=1;
    for(int i=2;i<=15;i++)
      f[i]=f[i-1]*10;
    for(int i=0;i<=9;i++)
      a[1][i].s[i]=1;
    for(int i=2;i<=12;i++)
      for(int j=0;j<=9;j++)
        for(int k=0;k<=9;k++)
          {
            a[i][k]=a[i][k]+a[i-1][j];
            a[i][k].s[k]+=f[i-1];
          }
    sum1=suan(r);
    sum2=suan(l-1);
    for(int i=0;i<9;i++)
      printf("%lld ",sum1.s[i]-sum2.s[i]);
    printf("%lld
",sum1.s[9]-sum2.s[9]);
    return 0;
}

数位DP，先预处理，a[i][j]代表有i位，最高位为j的各数字和。