bzoj1833: [ZJOI2010]count 数字计数 数位dp

传统数位dp思想，分位统计f[i][j][k]表示一个长i位的数第一位为j 到0 中字符k出现的次数。

#include <iostream>#include <cstring>#include <algorithm>#include <cstdio>#include <cmath>#include <queue>using namespace std;long long f[13][10][10],fac[20];struct node{    long long a[10];}t1,t2;void init(){    fac[1]=1;for(int i=2;i<=13;i++) fac[i]=fac[i-1]*10;    for(int i=0;i<=9;i++) f[1][i][i]=1;    for(int i=2;i<=12;i++)    {        for(int j=0;j<=9;j++)        {            for(int k=0;k<=9;k++)            {                for(int p=0;p<=9;p++)                {                    f[i][j][p]+=f[i-1][k][p];                }                f[i][j][j]+=fac[i-1];            }        }    }}node get(long long x){    node ans;for(int i=0;i<=9;i++)ans.a[i]=0;    if(!x){ans.a[0]=1;return ans;}    int len=1;    for(;x>=fac[len];len++);    len--;    for(int i=1;i<len;i++)    {        for(int j=1;j<=9;j++)        {            for(int k=0;k<=9;k++)            {                ans.a[k]+=f[i][j][k];            }        }    }    ans.a[0]++;    int cur=x/fac[len];for(int i=1;i<cur;i++)        for(int j=0;j<=9;j++)            ans.a[j]=ans.a[j]+f[len][i][j];x%=fac[len];ans.a[cur]+=x+1;    for(int i=len-1;i;i--)    {        cur=x/fac[i];        for(int k=0;k<cur;k++)        for(int j=0;j<=9;j++)            ans.a[j]=ans.a[j]+f[i][k][j];        x%=fac[i];ans.a[cur]+=x+1;    }    return ans;}int main(){    init();    long long a,b;    cin>>a>>b;    t1=get(b);    t2=get(a-1);    for(int i=0;i<=9;i++){printf("%lld",t1.a[i]-t2.a[i]);if(i!=9)printf(" ");}    return 0;}