BZOJ1833: [ZJOI2010]count 数字计数 (数位dp)

传送门

数位dp... ...大概都是这个套路吧... ...

写这个的时候直接水了一发... ...我也不知道自己写的是不是dp... ...

大概是主要内容和dp关系不大的dp... ...

mark代码..细长的代码真是丑啊..换行太频繁了....

#include<cmath>
#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
#include<queue>
using namespace std;
long long a,b;
long long f[15][5]={};
int shu[15]={};
long long zz[15]={};
long long ans[15]={};
void doit(long long x,long long fu){
    memset(shu,0,sizeof(shu));
    int k=0;
    long long y=x;
    while(y){
        shu[++k]=y%10;
        y/=10;
    }
    long long t;
    for(int i=1;i<=9;i++){
        for(int j=k;j>1;j--){
            if(shu[j]>i){
                t=shu[j]*f[j-1][1]+zz[j];
            }
            else{
                t=shu[j]*f[j-1][1];
            }
            ans[i]+=fu*t;
            if(shu[j]==i){
                ans[i]+=(x%zz[j]+1)*fu;
            }
        }
        if(shu[1]>=i){
            ans[i]+=fu;
        }
    }
    
    for(int j=k;j>1;j--){
        if(j==k){
            t=(shu[j]-1)*f[j-1][1];
        }
        else if(shu[j]>0){
            t=shu[j]*f[j-1][1]+zz[j];
        }
        else{
            t=shu[j]*f[j-1][1];
        }
        ans[0]+=fu*t;
        if(shu[j]==0){
            ans[0]+=(x%zz[j]+1)*fu;
        }
    }
    if(shu[1]>=0){
        ans[0]+=fu;
    }
    if(k>1){
        ans[0]+=fu*f[k-1][0];
    }
}
int main(){
    long long t=1;
    f[0][0]=1;
    for(int i=1;i<=13;i++){
        f[i][1]=f[i-1][1]*10+t;
        f[i][0]=f[i-1][0]+f[i-1][1]*9;
        zz[i]=t;
        t*=10;
    }
    scanf("%lld%lld",&a,&b);
    doit(b,1);
    doit(a-1,-1);
    for(int i=0;i<9;i++){
        printf("%lld ",ans[i]);
    }
    printf("%lld\n",ans[9]);
    return 0;
}