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

A_zjzj

已于 2022-06-03 17:59:04 修改

阅读量257

点赞数 1

分类专栏： # 【动态规划】数位dp 文章标签：数位dp

于 2020-12-10 20:03:20 首次发布

本文链接：https://blog.csdn.net/A_zjzj/article/details/110972847

版权

【动态规划】数位dp 专栏收录该内容

2 篇文章 0 订阅

订阅专栏

这道题其实就是一个数位 $d p$ 裸题，就是他这个前缀 $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;
}