[bzoj1833][ZJOI2010][count] (数位dp)

Description

给定两个正整数a和b，求在[a,b]中的所有整数中，每个数码(digit)各出现了多少次。

Input

输入文件中仅包含一行两个整数a、b，含义如上所述。

Output

输出文件中包含一行10个整数，分别表示0-9在[a,b]中出现了多少次。

Sample Input

Sample Output

HINT

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

Solution

十进制数位dp

设f(i,j,k)表示第i位以j开头的数含有多少个数字k

先预处理出f[i][j][k]数组

转移的时候从高到低转移

#include<stdio.h>
#define LL unsigned long long
int len,zt[];
LL f[][][],bt[],a,b,ans[];
void init() {
    bt[]=;
    for(int i=; i<=; i++)
        bt[i]=bt[i-]*;
    for(int i=; i<; i++)
        f[][i][i]=;
    for(int i=; i<=; i++)
        for(int j=; j<; j++)
            for(int k=; k<; k++) {
                for(int l=; l<; l++)
                    f[i][j][l]+=f[i-][k][l];
                f[i][k][k]+=bt[i-]; } }
void solve(LL x,LL d) {
    LL bef=x;
    for(len=; x; x/=)
        zt[++len]=x%;
    for(int i=; i<len; i++)
        for(int j=; j<; j++)
            for(int k=; k<; k++)
                ans[k]+=d*f[i][j][k];
    for(int i=len; i; i--) {
        for(int j=; j<zt[i]; j++) {
            if(!j && i==len)continue;
            for(int k=; k<; k++)
                ans[k]+=d*f[i][j][k]; }
        ans[zt[i]]+=d*(bef%bt[i]+); } }
int main() {
    init();
    scanf("%lld%lld",&a,&b);
    solve(b,);
    solve(a-,-);
    for(int i=; i<; i++)
        printf("%lld ",ans[i]);
        printf("%lld\n",ans[]);
    return ; }