说明
采用数位DP的方式来解题
(看了许多大佬的代码和解法,但是还是对于数位DP一知半解,还需要多多学习,希望大家多多指教)
解法
区间[A,B]
使用数位DP枚举出0–A-1中数字i出现的次数x,0–B中数字i出现的次数y
数字i在区间中出现的次数为y-x
#include<iostream>
#include<math.h>
using namespace std;
typedef long long ll;
ll dp[20];
void init(){
for(int i=1;i<20;i++){
dp[i]=i*pow(10,i-1);
}
}
ll solve(ll n,int x){
int len=0;//计算n的位数
ll digit=0,tail=0,radix=1,ans=0,cnt=n;
while(n){
digit=n%10;
n/=10;
len++;
//分3种情况讨论
if(digit>x) ans+=radix+digit*dp[len-1];
else if(digit==x)ans+=tail+1+digit*dp[len-1];
else{
ans+=digit*dp[len-1];
}
tail+=radix*digit;//尾数
radix*=10;
}
//要去掉首位为0的情况
if(x==0){
ll m=1;
while(cnt){
ans-=m;
m*=10;
cnt/=10;
}
}
return ans;
}
int main(){
ll a,b;
cin>>a>>b;
init();
for(int i=0;i<10;i++)
cout<<solve(b,i)-solve(a-1,i)<<endl;
}
简单范例比较(4位数范例)
for(int i=1;i<5;i++)
{
dp[i]=i*pow(10,i-1);
}
for(int i=0;i<10000;i++){
for(int j=0,x=i;j<4;j++){
a[x%10]++,x/=10;
}
}
//dp[4]与任意一个a中元素的数量是相同的为4000
//但是这个地方没有去掉为首位为0的情况
//延申的话要去掉首位为0的情况
首位为0的情况列举
位数 举例
1 0(1)
2 00 01 02 03 04 …(1+10)
3 000 001…(1+10+100)
…
n (1+10+100+…10^n-1)
运用以上就可以去掉首位为零的情况了