题目描述
给定两个正整数a和b,求在[a,b]中的所有整数中,每个数码(digit)各出现了多少次。
输入输出格式
输入格式:输入文件中仅包含一行两个整数a、b,含义如上所述。
输出文件中包含一行10个整数,分别表示0-9在[a,b]中出现了多少次。
最简单的数位DP的题目,可以对每个数字单独做,拿数字1举例。
考虑DP[x][high][limit][lead],表示最高位为high的x位数中1出现的次数,其中limit表示该位是否受限,若受限,下一位的取值不能大于给定数字的第x-1位a[x-1],lead表示是否有前导0
可以算出DP[x][high][limit][lead] = sum(DP[x-1][k][limit&&k == a[x-1]][lead && k == 0]) + cal(x,high,limit,lead),若limit,k的取值从0到a[x-1],否则k的取值从0到9。cal(x,high,limit,lead)计算了最高位high对答案的贡献,显然当high不为1的时候贡献为0,为1的时候计算结果为后x-1位在limit条件下的数的数量。
注意一下各种边界条件
代码:
#include <iostream>
#include <cstring>
#include <cstdio>
using namespace std;
long long dp[20][10][2][2];
long long pow10[] = {1,10,100,1000,10000,100000,1000000,10000000,1e8,1e9,1e10,1e11,1e12,1e13};
int a[20];
long long cal(int x,int high,bool limit,bool lead,int d){
long long ans = 0;
if(high != d || (lead && d == 0)) return 0;
if(x == 0) return 1;
if(!limit) return pow10[x];
x--;
while(x >= 0){
ans *= 10;
ans += a[x--];
}
return ans + 1;
}
long long solve(int x,int high,bool limit,bool lead,int d){
if(dp[x][high][limit][lead] != -1) return dp[x][high][limit][lead];
if(x == 0) return dp[x][high][limit][lead] = high == d ? 1 : 0;
long long tmp = 0;
if(limit){
for(int k = 0;k <= a[x-1];k++)
tmp += solve(x-1,k,k == a[x-1],lead && k == 0,d);
} else{
for(int k = 0;k < 10;k++)
tmp += solve(x-1,k,false,lead && k== 0,d);
}
tmp += cal(x,high,limit,lead,d);
return dp[x][high][limit][lead] = tmp;
}
long long solve_problem(long long x,int d){
if(x == 0) return d == 0 ? 1 : 0;
int len;
for(len = 0;x;len++){
a[len] = x % 10;
x /= 10;
}
memset(dp,-1,sizeof(dp));
return solve(len,0,true,true,d);
}
int main(){
long long l,r;
// while(1){
cin>>l>>r;
for(int i = 0;i < 10;i++){
cout<<(solve_problem(r,i)-solve_problem(l-1,i))<<" ";
}
cout<<endl;
// }
}