给定两个正整数a和b,求在[a,b]中的所有整数中,每个数码(digit)各出现了多少次。
这道题是一道比较好的数位dp,首先不要想着一下子解决,而要把大问题分解成小问题,1~9分别解决。定义f[i][j][k]意为i为枚举到现在的位数(从后往前),j为当前需算出的数码的值,k为当前已经算出的数码的个数,然后就是常规了。注意long long(本题一个大坑点),还有输出格式问题,前导0问题。
#include<cstdio>
#include<cstring>
#include<cstdlib>
#include<cmath>
#include<algorithm>
using namespace std;
int a[15];
long long f[15][11][15];
long long dfs(int pos,int sta,int he,bool lead,bool limt)
{
if(pos==0)return he;
if(lead==false && limt==false && f[pos][sta][he]!=-1)return f[pos][sta][he];
long long up=9,ans=0;
if(limt==true)up=a[pos];
for(int i=0;i<=up;i++)
{
int n1=he;bool bk1=false,bk2=false;
if(i==sta)n1++;
if(lead==true && i==0)
{
bk1=true;
if(sta==0)n1--;
}
if(limt==true && i==a[pos])bk2=true;
ans+=dfs(pos-1,sta,n1,bk1,bk2);
}
if(limt==false && lead==false)f[pos][sta][he]=ans;
return ans;
}
long long solve(long long x,int y)
{
int pos=0;
while(x)
{
a[++pos]=x%10;
x/=10;
}
return dfs(pos,y,0,true,true);
}
int main()
{
memset(f,-1,sizeof(f));
long long n,m;
scanf("%lld%lld",&n,&m);
for(int i=0;i<9;i++)printf("%lld ",solve(m,i)-solve(n-1,i));
printf("%lld\n",solve(m,9)-solve(n-1,9));
return 0;
}