洛谷 P2602 [ZJOI2010]数字计数 题解--zhengjun

这道题其实就是一个数位 \(dp\) 裸题，就是他这个前缀 \(0\) 有一点难处理，所以就多开一维就行了。

代码

#include<cstdio>
#include<cstring>
#define ll long long
using namespace std;
ll f[13][2][2][13];
int a[13];
ll dfs(int i,int is,int zero,int num,int sum){
    if(!i)return sum;
    if(f[i][is][zero][sum])return f[i][is][zero][sum];
    ll ans=0;
    for(int j=0;j<=(is?a[i]:9);j++){
        if(j==num&&(j!=0||!zero))ans+=dfs(i-1,is&(j==a[i]),zero&(j==0),num,sum+1);
        else ans+=dfs(i-1,is&(j==a[i]),zero&(j==0),num,sum);
    }
    return f[i][is][zero][sum]=ans;
}
ll get(ll x,int num){
    memset(f,0,sizeof(f));
    int k=0;
    while(x){
        a[++k]=x%10;
        x/=10;
    }
    return dfs(k,1,1,num,0);
}
int main(){
    ll a,b;
    scanf("%lld%lld",&a,&b);
    for(int i=0;i<=9;i++)
    printf("%lld ",get(b,i)-get(a-1,i));
    return 0;
}