【BZOJ-1833】count数字计数 数位DP

#include<iostream> 
#include<cstdio> 
#include<algorithm> 
#include<cstring> 
#include<cmath> 
using namespace std; 
long long L,R; 
long long F[15][10][10],ans1[10],ans2[10]; 
void prework() 
{ 
    for (int i=0; i<=9; i++) F[1][i][i]=1; 
    long long tmp=1; 
    for (int i=2; i<=12; i++) 
        { 
            tmp*=10; 
            F[i][0][0]=F[i-1][1][0]*9+F[i-1][0][0]+tmp; 
            for (int j=1; j<=9; j++)  
                F[i][0][j]=F[i-1][0][j]*9+F[i-1][j][j]; 
            for (int j=1; j<=9; j++) 
                { 
                    F[i][j][0]=F[i-1][1][0]*9+F[i-1][0][0]; 
                    for (int k=1; k<=9; k++) 
                        if (j==k) 
                            F[i][j][k]=F[i-1][0][k]*9+F[i-1][k][k]+tmp; 
                        else 
                            F[i][j][k]=F[i-1][0][k]*9+F[i-1][k][k]; 
                } 
        } 
} 
long long cf(int x) 
{ 
    long long re=1; 
    for (int i=1; i<x; i++) 
        re*=10; 
    return re; 
} 
void Calc(long long x,long long *ans) 
{ 
    int digit[15]={0},len=0; long long y=x; 
    while (x) {digit[++len]=x%10; x/=10;} 
    for (int i=1; i<len; i++) 
        for (int j=1; j<=9; j++) 
            for (int k=0; k<=9; k++) 
                ans[k]+=F[i][j][k]; 
    for (int i=len; i>=1; i--) 
        { 
            for (int j=0; j<=digit[i]-1; j++) 
                { 
                    if (i==len && j==0) continue; 
                    for (int k=0; k<=9; k++) ans[k]+=F[i][j][k]; 
                } 
            ans[digit[i]]+=y%cf(i)+1;            
        } 
} 
int main() 
{ 
    prework(); 
    scanf("%lld%lld",&L,&R); 
    Calc(L-1,ans1); Calc(R,ans2); 
    printf("%lld",ans2[0]-ans1[0]); 
    for (int i=1; i<=9; i++) printf(" %lld",ans2[i]-ans1[i]); 
    return 0; 
}