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

数位dp的套路记忆化搜索

枚举d=0~9，分别统计

f[pos][sum]表示当前在pos位，d这个数出现个数

flag表示是否限位，k表示是否前导0

 #include<iostream>
 #include<cstdio>
 #include<cstring>
 #include<algorithm>
 #include<cmath>
 using namespace std;
 typedef long long lol;
 int s[],len;
 lol A,B,f[][];
 lol dfs(int pos,int sum,int flag,int k,int d)
 {int i;
   lol cnt=;
   if (pos<=) return sum;
   if (!flag&&k&&f[pos][sum]!=-) return f[pos][sum];
   int ed=;
   if (flag) ed=s[pos];
   for (i=;i<=ed;i++)
     {
       if (k==&&i==)
     cnt+=dfs(pos-,sum,flag&&(i==ed),,d);
       else
     {
       cnt+=dfs(pos-,sum+(i==d),flag&&(i==ed),,d);
     }
     }
   if (!flag&&k) f[pos][sum]=cnt;
   return cnt;
 }
 lol solve(lol x,int d)
 {int i,j,k;
   memset(f,-,sizeof(f));
   len=;
   while (x)
     {
       s[++len]=x%;
       x/=;
     }
   return dfs(len,,,,d);
 }
 int main()
 {int i;
   cin>>A>>B;
   printf("%lld",solve(B,)-solve(A-,));
   for (i=;i<=;i++)
     {
       printf(" %lld",solve(B,i)-solve(A-,i));
     }
 }