码迷,mamicode.com
首页 > 其他好文 > 详细

BZOJ_1833_[ZJOI2010]count 数字计数_数位DP

时间:2018-03-04 23:56:48      阅读:179      评论:0      收藏:0      [点我收藏+]

标签:ext   ring   cursor   图片   数字   icons   代码   scan   span   

BZOJ_1833_[ZJOI2010]count 数字计数_数位DP

题意:

给定两个正整数a和b,求在[a,b]中的所有整数中,每个数码(digit)各出现了多少次。

 

分析:

数位DP

f[i][j][k]表示i位数,以j开头的数中k出现的次数

预处理出来10的幂(在数位DP中经常会用到)

f[i][j][k]+=f[i-1][l][k]+(j==k)*10^(i-1)

之后按位枚举,0的情况特殊处理

 

代码:

#include <stdio.h>
#include <string.h>
#include <algorithm>
using namespace std;
#define LL long long
LL f[15][11][11],a,b,mi[15];
void init(){
    mi[1]=1;
    for(int i=2;i<=13;i++){
        mi[i]=mi[i-1]*10;
    }
    for(int i=0;i<=9;i++){
        f[1][i][i]=1;
    }
    for(int i=2;i<=13;i++){
        for(int j=0;j<=9;j++){
            for(int k=0;k<=9;k++){
                for(int l=0;l<=9;l++){
                    f[i][j][k]+=f[i-1][l][k];
                    if(j==k)f[i][j][k]+=mi[i-1];
                }
            }
        }
    }
}
LL calc(LL x,int p){
    if(!x)return (!p);
    LL re=0;
    int d=13;
    while(mi[d]>x)d--;
    //d++;
    for(int i=1;i<d;i++){
        for(int j=1;j<=9;j++){
            re+=f[i][j][p];
        }
    }
    if(!p)re++;
    int cur=x/mi[d];
    for(int i=1;i<cur;i++){
            re+=f[d][i][p];
    }
    x%=mi[d];
    if(cur==p)re+=x+1;
    for(int i=d-1;i;i--){
        cur=x/mi[i];
        for(int j=0;j<cur;j++){
            re+=f[i][j][p];
        }
        x%=mi[i];
        if(cur==p)re+=x+1;
    }
    return re;
}
int main(){
    scanf("%lld%lld",&a,&b);
    init();
    int flg=0;
    for(int i=0;i<=9;i++){
        if(!flg)flg=
            printf("%lld",calc(b,i)-calc(a-1,i));
        else printf(" %lld",calc(b,i)-calc(a-1,i));
    }
}

 

技术分享图片

BZOJ_1833_[ZJOI2010]count 数字计数_数位DP

标签:ext   ring   cursor   图片   数字   icons   代码   scan   span   

原文地址:https://www.cnblogs.com/suika/p/8506697.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!