题目:给定两个正整数a和b,求在[a,b]中的所有整数中,每个数码(digit)各出现了多少次。
思路:数位dp,sum表示digit出现的次数,num表示有多少个数。前导0需要注意下。
#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
int wei[15],k;
ll a,b,ans[15],p[15];
struct node{
ll sum,num;
};
node dp[15];
node dfs(int pos,int limit,int lead)
{
if(pos<1)
{
node fuck;
fuck.num=1;
fuck.sum=0;
return fuck;
}
if(!limit&&!lead&&dp[pos].sum!=-1)
return dp[pos];
int up=limit?wei[pos]:9;
node ans,tmp;
ans.sum=ans.num=0;
for(int i=0;i<=up;i++)
{
if(i==k)
{
if(lead&&k==0)
{
tmp=dfs(pos-1,limit&&i==up,lead&&i==0);
ans.sum+=tmp.sum;
ans.num+=tmp.num;
continue;
}
if(limit&&i==up)
{
tmp=dfs(pos-1,limit&&i==up,lead&&i==0);
ans.sum+=tmp.sum+tmp.num;
ans.num+=tmp.num;
continue;
}
tmp=dfs(pos-1,limit&&i==up,lead&&i==0);
ans.sum+=tmp.sum+tmp.num;
ans.num+=tmp.num;
continue;
}
tmp=dfs(pos-1,limit&&i==up,lead&&i==0);
ans.num+=tmp.num;
ans.sum+=tmp.sum;
}
if(!limit&&!lead)
dp[pos]=ans;
return ans;
}
ll solve(ll x)
{
memset(dp,-1,sizeof dp);
int len=0;
while(x)
{
wei[++len]=x%10;
x/=10;
}
node ans=dfs(len,1,1);
return ans.sum;
}
int main()
{
p[0]=1;
for(int i=1;i<=13;i++)
p[i]=p[i-1]*10;
scanf("%lld%lld",&a,&b);
for(k=0;k<=9;k++)
ans[k]=solve(b)-solve(a-1);
for(int i=0;i<9;i++)
printf("%lld ",ans[i]);
printf("%lld\n",ans[9]);
return 0;
}