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

Description
给定两个正整数a和b，求在[a,b]中的所有整数中，每个数码(digit)各出现了多少次。
Input
输入文件中仅包含一行两个整数a、b，含义如上所述。
Output
输出文件中包含一行10个整数，分别表示0-9在[a,b]中出现了多少次。
Sample Input
1 99

Sample Output
9 20 20 20 20 20 20 20 20 20

HINT
30%的数据中，a<=b<=10^6；
100%的数据中，a<=b<=10^12。

传送门
靠……一开始一直以为记忆化是gg的于是搁置了这么久……

很裸的数位dp，我写的是记忆化……（怎么搞都可以啦）
可以枚举0~9，然后分别统计[1,B],[1,A)。
……这东西还真没什么好说的= =
为了省点memset的时间我f数组开了4维（其实本质一样的）
……会写dfs就会这题啦！
代码里可能有些书写技巧。。具体看代码吧！

#include<bits/stdc++.h>
#define ll long long
using namespace std;
int len,a[15];
ll ans,NUM[10];
ll dp[13][10][2][13];
ll dfs(int now,int x,bool limit,bool f,ll sum){
    if (!now) return sum;
    if (!limit && dp[now][x][f][sum]) return dp[now][x][f][sum];
    int t,tmp;
    if (!limit) t=9;else t=a[now];
    ll tt=0LL;
    for (int i=0;i<=t;i++){
        if (i==0 && f) tmp=0; else tmp=1;
        tt+=dfs(now-1,x,limit&(i==a[now]),f&(i==0),sum+(i==x)*tmp);
    }
    if (!limit) dp[now][x][f][sum]=tt;
    return tt;
}
void get(ll x){
    len=0;
    while (x) a[++len]=(x%10),x/=(ll)10; 
}
void ANS(ll x,bool f){
    get(x);
    memset(dp,0,sizeof(dp));
    for (int i=0;i<=9;i++){
        ans=dfs(len,i,1,1,0);
        if (!f) NUM[i]+=ans;
         else NUM[i]-=ans;
    }
}
int main(){
    ll A,B;
    cin>>A>>B;
    ANS(B,0),ANS(A-1,1);
    for (int i=0;i<9;i++) printf("%lld ",NUM[i]);
    printf("%lld\n",NUM[9]);
    return 0;
}