Bzoj1833 [ZJOI2010]count 数字计数

1833: [ZJOI2010]count 数字计数

Time Limit: 3 Sec  Memory Limit: 64 MB

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。

 

Solution：

　　一道不算特别裸的数位dp，需要预处理一下每一个长度所包含的数字个数，并且前导零很恶心。

 

#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>
using namespace std;
long long  l,r,bit[15],f[15],c[15],p[15];
long long dfs(int pos,int pre,int Dig,bool Limit) {
    if(pos<=0) return 0;
    if(pre>=0&&!Limit&&f[pos]!=-1) return f[pos];
    int End=Limit?bit[pos]:9;
    long long ans=0;
    for(int i=0; i<=End; ++i) {
        if(pre<0&&!i) ans+=dfs(pos-1,pre,Dig,Limit&&i==End);
        else if(i==Dig) {
            if(i==End&&Limit) ans+=c[pos-1]+1+dfs(pos-1,i,Dig,Limit&&i==End);
            else ans+=p[pos-1]+dfs(pos-1,i,Dig,Limit&&i==End);
        } else ans+=dfs(pos-1,i,Dig,Limit&&i==End);
    } if(pre>=0&&!Limit) f[pos]=ans;
    return ans;
}
long long Get(long long x,int dig) {
    memset(f,0xff,sizeof(f));
    memset(bit,0,sizeof(bit));
    while(x) bit[++bit[0]]=x%10,x/=10,c[bit[0]]=c[bit[0]-1]+bit[bit[0]]*p[bit[0]-1];
    return dfs(bit[0],-10,dig,true);
}
int main() {
    scanf("%lld%lld",&l,&r); --l;
    p[0]=1; for(int i=1; i<15; ++i) p[i]=p[i-1]*10;
    for(int i=0; i<=9; ++i) {
        printf("%lld",Get(r,i)-Get(l,i));
        if(i<9) putchar(' ');
    } return 0;
}