HDU 3652 B-number(数位dp&记忆化搜索)

Posted shiyicode

tags:

篇首语:本文由小常识网(cha138.com)小编为大家整理,主要介绍了HDU 3652 B-number(数位dp&记忆化搜索)相关的知识,希望对你有一定的参考价值。

题目链接[kuangbin带你飞]专题十五 数位DP G - B-number

题意

求1~n的范围里含有13且能被13整除的数字的个数。

思路

首先,了解这样一个式子:a%m == ((b%m)*c+d)%m;
式子的正确是显然的,就不证明了。
那么判断数是否可以被13整除就可以分解为一位一位进行处理。
当然,我们也只需要储存取余后的值。
dfs(len, num, mod, flag)
mod记录数字对13取余后的值
len表示当前位数
num==0 不含13且上一位不为1
pre==1 不含13且上一位为1
pre==2 含13
flag表示是否可以任意取值(判断范围)。
如此,记忆化搜索即可得解。

代码

#include <iostream>
#include <algorithm>
#include <cstring>
#include <cstdio>
#include <cstdlib>
#include <vector>

using namespace std;

#define LL long long
#define MOD 13
LL dp[20][3][13];
int dis[20];

LL dfs(int len, int type, int mod, bool flag)
{
    if(len < 0)
        return type == 2 && mod == 0;
    if(!flag && dp[len][type][mod]!=-1)
        return dp[len][type][mod];
    int end = flag?dis[len]:9;
    int ans = 0;
    for(int i=0; i<=end; i++)
    {
        if(type == 2 || (type == 1 && i == 3))
            ans += dfs(len-1, 2, (mod*10+i)%MOD, flag&&i==end);
        else
            ans += dfs(len-1, i==1?1:0, (mod*10+i)%MOD, flag&&i==end);
    }
    if(!flag)
        dp[len][type][mod] = ans;
    return ans;
}

LL solve(LL n)
{
    int len = 0;
    while(n)
    {
        dis[len++] = n%10;
        n /= 10;
    }
    return dfs(len-1, 0, 0, 1);
}

int main()
{
    int n;
    memset(dp, -1, sizeof(dp));
    while(cin>>n)
        cout<<solve(n)<<endl;
    return 0;
}

以上是关于HDU 3652 B-number(数位dp&记忆化搜索)的主要内容,如果未能解决你的问题,请参考以下文章

hdu-3652 B-number 数位DP

HDU - 3652 - B-number(数位DP)

HDU 3652 B-number 数位DP

HDU 3652 B-number (数位DP)

hdu3652B-number 数位DP

HDU 3652 B-number(数位DP)