[BZOJ5064]B-number

[BZOJ5064]B-number

题目大意:

\(1\sim n(n\le10^{15})\)间有多少数满足是\(13\)的倍数且包含字符串\(13\)

思路:

数位DP。\(f[i][j][k]\)表示考虑\(i\)位,模\(13\)余数是\(j\),上一位数字是\(k\)时的方案数。

源代码:

#include<cstdio>
#include<cctype>
#include<utility>
#include<cstring>
typedef long long int64;
inline int64 getint() {
    register char ch;
    while(!isdigit(ch=getchar()));
    register int64 x=ch^'0';
    while(isdigit(ch=getchar())) x=(((x<<2)+x)<<1)+(ch^'0');
    return x;
}
int d[16];
int64 pwr[16];
typedef std::pair<int64,int64> Node;
Node f[16][13][10];
Node dp(const int &dep,const bool &zero,const bool &limit,const int &r,const int &last) {
    if(dep==0) return std::make_pair(!zero&&!r,0);
    const int lim=limit?d[dep]:9;
    if(!zero&&!limit&&f[dep][r][last]!=(Node){-1,-1}) return f[dep][r][last];
    Node ret=std::make_pair(0,0);
    for(register int i=0;i<=lim;i++) {
        const Node p=dp(dep-1,zero&&!i,limit&&i==lim,(r+i*pwr[dep])%13,i);
        if(last==1&&i==3) {
            ret.second+=p.first;
        } else {
            ret.first+=p.first;
        }
        ret.second+=p.second;
    }
    if(!zero&&!limit) f[dep][r][last]=ret;
    return ret;
}
inline int64 solve(int64 x) {
    for(;x;x/=10) {
        d[++d[0]]=x%10;
        pwr[d[0]]=d[0]==1?1:pwr[d[0]-1]*10%13;
    }
    return dp(d[0],true,true,0,0).second;
}
int main() {
    memset(f,-1,sizeof f);
    const int64 n=getint();
    printf("%lld\n",solve(n));
    return 0;
}

猜你喜欢

转载自www.cnblogs.com/skylee03/p/9917522.html