数位DP—计数问题

问题链接： AcWing 338. 计数问题
问题描述

分析

数据范围1e8，从小到大枚举计数肯定会超时
换一种解体思路，求[a,b]之间的数字个数就是求[1,b]-[1,a-1]的数字个数，所以我们写一个count数组，求[1,b]
中的0~9数字个数即可，具体求法如下：
假设b=abcdef,要求[1,abcdef]中数字1的个数(同理能求1就能求2~9的数字个数，0需要特判，因为有前导0不合法)
当b=abcdef时，分类讨论数字1在位置4时能构成的数字个数
(1)xxx1yyy, xxx<abc时，即0~abc-1,yyy可以任取，此时的方案数为(abc-1+1)*(999+1)=abc*1000
(2)xxx1yyy, xxx=abc时,此时讨论1与d的大小关系
	(2.1)d<1,此时无合法的情况 0种
	(2.2)d==1,此时yyy的取值范围为0~def def+1种
	(2.3)d>1,此时yyy可以任取[0~999]，1000种
注意求数字0时需要特判，首先0不能出现在最高位，也就是a=0的情况
其次，当求a时，xxx是从001开始，不是从000，例如000 0 123不是合法数字

代码如下

#include<cstdio>
#include<iostream>
#include<algorithm>
#include<cstring>
#include<vector>
using namespace std;

int getpow10(int x){
    int res=1;
    while(x--) res*=10;
    return res;
}
int getnum(vector<int> v,int l,int r){
    int res=0;
    for(int i=r;i>=l;i--) res=res*10+v[i];
    
    return res;
}
int count(int x,int t){
    int res=0;
    vector<int> v;
    while(x){
        v.push_back(x%10);
        x/=10;
    }
    for(int i=v.size()-1-!t;i>=0;i--){
        if(i<v.size()-1){
            res+=getnum(v,i+1,v.size()-1)*getpow10(i);
            if(!t) res-=getpow10(i);
        }
        if(v[i]==t) res+=getnum(v,0,i-1)+1;
        else if(v[i]>t) res+=getpow10(i);
    }
    return res;
}
int main(){
    int a,b;
    while(cin>>a>>b){
        if(a==0&&b==0) break;
        if(a>b) swap(a,b);
        for(int i=0;i<10;i++)
            cout<<count(b,i)-count(a-1,i)<<" ";
        cout<<endl;
    }
    return 0;
}