计数器（区间dp模板）

计数器

题目描述

一本书的页数为 $n$，页码从 $1$ 开始编起，请你求出全部页码中，用了多少个 $0,1,2\cdots 9$。每个页码不含前导 $0$，如 $n=1234$ 时第 $5$ 页不是 $0005$，只是 $5$。

输入格式

一个正整数 $n$，表示总的页码。

输出格式

输出共十行，其中第 $k$ 行为数字 $k-1$ 的个数。

样例 #1

样例输入 #1

11

样例输出 #1

1
4
1
1
1
1
1
1
1
1

提示

对于 $100\%$ 数据，满足 $1\le n\le 10^9$。

---

$\mathrm{2022.7.4}$：新添加三组 $\text{Hack}$ 数据。

模板

//数位dp

#include<iostream>
#include<algorithm>
#include<cstring>
#include<vector>
#include<cmath>

using namespace std;

const int N = 11;

int f[N][N][N];//表示最高位为j，已经有了i位数，当前位数的数是k的个数
int n;

void init(){
    for(int i=0;i<=9;i++){
        f[1][i][i]=1;
    }
    
    for(int i=2;i<=N;i++){//枚举位数
        for(int j=0;j<=9;j++){//枚举最高位数字
            for(int u=0;u<=9;u++){//枚举当前位数
                if(j==u)f[i][j][u]+=pow(10,i-1);//画个图就懂了（相对于后面的i-1位数，多了10^(i-1)的方案）
                //为什么呢？因为你j跟u相等，
                 /* 
                    例如计算位数为3(i)，且最高位的数为1的数的时候：

            1.那么先要加上100 ~ 199以内1的个数，即为pow(10,i - 1)个1
            2.注意不需要加上100以内1的个数,即分别是【[0,10],[11，20],...[90,99]】内1的个数，因为需要特判前导零，所以应该只计算位数为3的数里面含1的个数


                */
                for(int k=0;k<=9;k++){
                    f[i][j][u]+=f[i-1][k][u];
                }
                
            }
        }
    }
}

int count(int n,int u){
    if(!n)return u?0:1;
    
    vector<int>num;
    
    while(n){
        num.push_back(n%10);
        n/=10;
    }
    
    int res=0,last=0;//前面位数出现的u的个数
    
    
    for(int i=num.size()-1;i>=0;i--){
        int x=num[i];//每一位数的限制
        
        //统计后面u的个数
        //计算最高位填0 ~ x - 1的情况
        //记得最高位特判前导0的情况,需要从1开始（关键）
        for(int j=((i==num.size()-1)?1:0);j<x;j++){
            res+=f[i+1][j][u];
        }
        
        //统计前面u的个数，而且前面出现的次数已经固定了，所以只需要乘起来就是结果
        res+=last*x*pow(10,i);
        
        if(x==u)last++;
        
        //枚举到最后一个数字了 ，结果再加上最后一个数字含有的u的数量,即为last
        if(!i){
            res+=last;
        }
        
    }
    
    
    for(int i=1;i<num.size();i++){
        for(int j=i!=1;j<=9;j++){
            res+=f[i][j][u];
        }
    }
    
    return res;
}

int main(){
    init();
    cin>>n;
    
    for(int i=0;i<=9;i++){
        cout<<max(count(n,i)-count(0,i),0)<<endl;
    }
    
    return 0;
}