洛谷 P2602 [ZJOI2010]数字计数

题目描述

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

输入输出格式
输入格式：

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

输出格式：

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

输入输出样例

输入样例#1：
1 99

输出样例#1：
9 20 20 20 20 20 20 20 20 20

说明

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

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

---

【分析】
马丹预处理处理错了结果查了半天卡位…诶哟我真想rigel
dp[i][j][k]表示i位数字，开头为j，数字k出现的次数，然后就可以乱搞了

---

【代码】

//zjoi count#include<cstdio>#include<cstring>#include<iostream>#define ll long long#define fo(i,j,k) for(i=j;i<=k;i++)using namespace std;ll a,b;ll dp[15][10][10],digit[15],ans[2][10],pw[15];   //i位数，首位为j，数字k出现的次数 void work(ll n,int c){    ll i,j,k,x,y,len=0;    memset(digit,0,sizeof digit);    ll tmp=n;    while(n)      digit[++len]=n%10,n/=10;    int h=len-1;    while(h>=0)    {        tmp%=pw[h];        ans[c][digit[h+1]]+=tmp+1;        h--;    }    for(i=len-1;i>=1;i--)      fo(j,1,9) fo(k,0,9) ans[c][k]+=dp[i][j][k];    fo(j,1,digit[len]-1)      fo(k,0,9)        ans[c][k]+=dp[len][j][k];    for(i=len-1;i>=1;i--)      fo(j,0,digit[i]-1)        fo(k,0,9)          ans[c][k]+=dp[i][j][k];}int main(){//  freopen("countzj.in","r",stdin);//  freopen("countzj.out","w",stdout);    int i,j,k,x,y;    scanf("%lld%lld",&a,&b);pw[0]=1;    fo(i,1,12) pw[i]=pw[i-1]*10;    fo(i,0,9) dp[1][i][i]=1;    fo(i,2,12)      fo(j,0,9)        fo(k,0,9)        {            if(j==k) dp[i][j][k]+=pw[i-1];            fo(x,0,9)              dp[i][j][k]+=dp[i-1][x][k];        }    work(b,0);    work(a-1,1);    fo(i,0,9) printf("%lld ",ans[0][i]-ans[1][i]);    printf("\n");    return 0;}/* 100000 100001*/