非負整數區間內含某數字的數的個數

羣裏有人問了這個問題

而後被attack秒了orz

考慮數位DP

預處理v[第幾位][該位是不是目標數字][是否有目標數字]

顯然數字除了0以外的求解過程是徹底一致的

那麼咱們特判0就能夠了

#include"cstdio"
#include"cstring"
#include"iostream"
#include"algorithm"
using namespace std;

long long L,R,val;
long long v[19][2][2],pw[19],lis[19];

long long dfs(long long x,int ct,bool fl)
{
    if(!ct) return fl?x+1:x>=val;
    int w=x/pw[ct];long long sum=0;
    for(int i=0;i<w;++i) if(i!=val) sum+=v[ct-1][1][1]+v[ct-1][0][1]+(fl?v[ct-1][0][0]:0);
    if(w>val) sum+=pw[ct];
    return sum+dfs(x-pw[ct]*w,ct-1,fl||w==val);
}

long long slv(long long x)
{
    if(x<10) return x>=val;
    int ct=18;
    while(pw[ct]>x) --ct;
    int w=x/pw[ct];
    long long sum=(val?v[ct-1][1][1]:lis[max(ct-2,0)]+1)+v[ct-1][0][1];
    for(int i=1;i<w;++i) if(i!=val) sum+=v[ct-1][1][1]+v[ct-1][0][1];
    if(val&&w>val) sum+=pw[ct];
    return sum+dfs(x-pw[ct]*w,ct-1,w==val);
}

int main()
{
    pw[0]=1;v[0][1][1]=1,v[0][0][0]=9;
    for(int i=1;i<=18;++i) pw[i]=pw[i-1]*10;
    for(int i=1;i<=17;++i){
        v[i][0][1]+=9*(v[i-1][1][1]+v[i-1][0][1]);
        v[i][0][0]+=9*v[i-1][0][0];
        v[i][1][1]+=v[i-1][1][1]+v[i-1][0][1]+v[i-1][0][0];
        lis[i]=v[i][0][1]+lis[i-1];
    }scanf("%lld%lld%d",&L,&R,&val);--L;
    printf("%lld\n",slv(R)-slv(L));
    return 0;
}