1833: [ZJOI2010]count 数字计数

题目

bzoj1833
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。

题解

最近做数位DP的题总是出各种错误，调了好久才出来。数位DP就是这样，一点细节都不能错。

下面进入正题。

f[i][j]代表在无范围限制下填到第i位数码j出现的次数。
g[i][j]代表在不超过上界x的前提下填到第i位数码j出现的次数。
填到第i位时 cnt1:前导0的个数 cnt2:$10^{i-1}$ cnt3:s[len-i..len-1]所组成的数。
那么转移时

 cnt3=(s[len-i]-'0')* cnt2+cnt3; cnt2*=10; cnt1=cnt2+cnt1;

转移f时枚举第i为填j时

f[i][l]+=f[i-1][l] ,f[i][j]+=cnt2;(0<=j<=9,0<=l<=9)

转移g时先枚举第i为填j（0<=j<原数中该位）此时1-i-1位都没有限制，用f转移，第i为为原数中该位时就该用g来转移；也就是

g[i][l]+=f[i-1][l]，g[i][j]+=cnt2;（j< s[len-i-1]-'0'，0<=l<=9）
g[i][l]+=g[i-1][l], g[i][j]+=cnt3+1;(j=s[len-i-1]-'0'，0<=l<=9)

最后还要减去多算的前导0，即：

g[len-1][0]=g[len-1][0]-cnt1+1;

---

代码：

#include<cstdio>
#include<cstring>
#include<algorithm>
#include<iostream>
using namespace std;

typedef long long ll;
ll f[20][10],g[20][10],ans[10],l,r,cnt1,cnt2,cnt3; //cnt1:前导0的个数 cnt2:10^i-1 cnt3:s[len-i..len-1] 
char s[20];

void count(ll x,int k){
    sprintf(s,"%lld",x);
    memset(f,0,sizeof(f));
    memset(g,0,sizeof(g));
    int len=strlen(s);
    for(int i=0;i<10;i++)f[0][i]=1;
    for(int i=0;i<=s[len-1]-'0';i++)g[0][i]=1;
    cnt2=1; cnt1=1; cnt3=0;
    for(int i=1;i<len;i++){
        cnt3=(s[len-i]-'0')*cnt2+cnt3;
        cnt2*=10;
        cnt1=cnt2+cnt1;
        for(int j=0;j<10;j++){
            for(int l=0;l<10;l++)f[i][l]+=f[i-1][l];
            f[i][j]+=cnt2;
        }
        for(int j=0;j<s[len-i-1]-'0';j++){
            for(int l=0;l<10;l++)
            g[i][l]+=f[i-1][l];
            g[i][j]+=cnt2;
        }
        for(int j=0;j<10;j++)
        g[i][j]+=g[i-1][j];
        g[i][s[len-1-i]-'0']+=cnt3+1;
    }
    g[len-1][0]=g[len-1][0]-cnt1+1;
    for(int i=0;i<10;i++)ans[i]+=k*g[len-1][i];
}

int main(){
    scanf("%lld%lld",&l,&r);
    count(r,1);
    count(l-1,-1);
    for(int i=0;i<9;i++)printf("%lld ",ans[i]);
    printf("%lld\n",ans[9]);
    return 0;
}