bzoj 1833: [ZJOI2010]count 数字计数数位dp

Posted 2020-12-25 lokiii

非典型数位dp
先预处理出f[i][j][k]表示从后往前第i位为j时k的个数，然后把答案转换为ans(r)-ans(l-1)，用预处理出的f数组dp出f即可（可能也不是dp吧……）

#include<iostream>
#include<cstdio>
using namespace std;
long long l,r,t[25];
struct dp
{
    long long a[15];
    dp operator + (dp x)
    {
        dp r;
        for(int i=0;i<=9;i++)
            r.a[i]=a[i]+x.a[i];
        return r;
    }
}f[20][20];
dp work(long long x)
{
    dp ans;
    for(int i=0;i<=9;i++)
        ans.a[i]=0;
    if(!x)
    {
        ans.a[0]=1;
        return ans;
    }
    int len=15;
    while(t[len]>x)
        len--;
    for(int i=1;i<len;i++)
        for(int j=1;j<=9;j++)
            ans=ans+f[i][j];
    ans.a[0]++;
    int cur=x/t[len];
    for(int i=1;i<cur;i++)
        ans=ans+f[len][i];
    x%=t[len];
    ans.a[cur]+=x+1;
    for(int i=len-1;i;i--)
    {
        cur=x/t[i];
        for(int j=0;j<cur;j++)
            ans=ans+f[i][j];
        x%=t[i];
        ans.a[cur]+=x+1;
    }
    return ans;
}
int main()
{
    t[1]=1;
    for(int i=2;i<=15;i++)
        t[i]=t[i-1]*10;
    for(int i=0;i<=9;i++)
        f[1][i].a[i]=1;
    for(int i=2;i<=12;i++)
        for(int j=0;j<=9;j++)
            for(int k=0;k<=9;k++)
            {
                f[i][k]=f[i][k]+f[i-1][j];
                f[i][k].a[k]+=t[i-1];
            }
    scanf("%lld%lld",&l,&r);
    dp ans1=work(r),ans2=work(l-1);//cout<<"aa";
    for(int i=0;i<=9;i++)
        printf("%lld ",ans1.a[i]-ans2.a[i]);
    return 0;
}