【数位DP】数字统计

题目

给定两个正整数a和b,求在[a,b]中的所有整数中,每个数码(digit)各出现了多少次。

数位DP

(1)分情况,逐位讨论。

(2)模型:计算在[L,R]中有多少个数满足条件。

(3)套路:将问题转化为[1,R]-[1,L-1],只需回答[1,X]的询问即可。

思路

1.算出[1,X]:

(1)按位拆分,为后面做铺垫

(2)预处理:

	 1)pow[]表示10的次幂

	 2)f[]表示:后i位自由的话每个数字出现的次数

ps:有一个规律。0~9每个数字出现1次,10~99出现20次,100~999出现300次。出现次数为:n*10^(n-1)(n为位数)

(3)处理比X少一位的情况

       

 从0~9共10个数。同位数计算10次,加上pow[i-1]即可。

(4)处理全部位数

	从最高位到最低位。比如ABCDE,A会出现BCDE+1次,以此类推。

2.将第一遍算出的答案全部*(-1),就相当于减去。

#include<iostream>
using namespace std;
long long ans[10],f[20],pow[20];
int p[20],l;
void deal(long long x, long long y){ // 统计固定位是x,自由位有y个的情况 
    for(long long i=0;i<10;i++) ans[i]+=f[y]; // 算后面的 
    for(long long p=x;p;p/=10) ans[p%10]+=pow[y]; // 算前面的 
}
void calc(long long x){
    if(!x) return;
    l=0,pow[0]=1;
    for(long long tp=x;tp;tp/=10) p[++l]=tp%10; // 一位一位拆分 
    for(int i=1;i<=l;i++) pow[i]=pow[i-1]*10; // 10的次幂 
    for(int i=1;i<=l;i++) f[i]=pow[i-1]*i; // 后i位是自由的话,每个数字出现多少次 
    // 处理比l小位数的情况 
    for(int i=1;i<l;i++) // 枚举,当前是i位数 
        for(int k=1;k<10;k++) deal(k,i-1); // 枚举这个i位数,是以k开头的 
    // 处理l位数的情况
    // i : 当前算到哪个位置
    // lst : 当前前面的数是多少
    // st,en : 数字的枚举范围 
    for(long long i=l,lst=0,st,en;i;lst+=p[i--]){
        st=(i==l)? 1:0;//如果i==l,则这是最高位,不能是0 
        en=(i==1)? p[i]:p[i]-1;// 
        lst *= 10;
        for(int k = st; k <= en;k++) deal(lst+k, i-1);
    }
}
int main(){
    long long a,b;
    cin>>a>>b;
    calc(a-1);
    for(int i=0;i<10;i++) ans[i]*=-1;
    calc(b);
    for(int i=0;i<10;i++) cout<<ans[i]<<" ";
    return 0;
}

欢迎指正评论O(∩_∩)O~~

原文地址:https://www.cnblogs.com/kylinbalck/p/9923081.html