【数位dp】HDU - 3555 Bomb

Problem Description

输入T组测试数据，每组测试数据给你一个n，让你求1-n中有多少个数 数中包含有连续49,(1 <= n <= 2^63-1);

Sample Input

3
1
50
500

Sample Output

0
1
15

思路:

还在学习过程中
我分别用了两种解题思路：一种是参考博客，一种是问了东庆
第一种：dfs记忆化搜索过程中，搜索到49后，后面就不需要搜索了，直接计算。二维时间复杂度稍短
第二种：直接记忆化搜索。三维数组

第一种代码

#include<bits/stdc++.h>
using namespace std;
#define ll long long
ll dp[50][2], a[50];
ll z[50], n;
ll dfs(int pos, int sta, int limit)
{
    if(pos < 0) return 0;
    if(!limit && dp[pos][sta] != -1) return dp[pos][sta];
    int up = limit ? a[pos] : 9;
    ll res = 0;
    for(int i = 0; i <= up; i++)
    {
        if(sta && i == 9)
            res += limit ? (n%z[pos]+1) : z[pos];//如果找到49，如果没有限制，就是随便选。有限制，就看限制允许最大的个数。
        else res += dfs(pos-1, i == 4, limit && i == a[pos]);//没有找到49就继续记忆化搜索
    }
    if(!limit) return dp[pos][sta] = res;
    return res;
}
ll solve(ll n)
{
    int pos = 0;
    while(n)
    {
        a[pos++] = n%10;
        n /= 10;
    }
    return dfs(pos-1, false, true);
}
int main()
{
    int T;
    scanf("%d", &T);
    memset(dp, -1, sizeof(dp));
    z[0] = 1;
    for(int i = 1; i <= 18; i++)//预处理求出 给你i个位置，0-9随便填，有多少种情况
        z[i] = z[i-1] * 10;
    while(T--)
    {
        cin >> n;
        cout << solve(n) << endl;
    }
    return 0;
}

第二种

#include<bits/stdc++.h>
using namespace std;
#define ll long long
ll a[50], dp[50][15][2];//三维数组,分别代表当前剩余几位没搜索，前一位是多少，当前是否包含49
ll dfs(int pos, int pre, int sta, int limit)
{
    if(pos == -1) return sta;//如果搜索完一个完整的数，包含49返回1，否则返回0
    if(!limit && dp[pos][pre][sta] != -1) return dp[pos][pre][sta];
    int up = limit ? a[pos]:9;
    ll res = 0;
    for(int i = 0; i <= up; i++)
    {
        if(pre == 4 && i == 9)//如果找到49，sta = 1.否则sta不变。
            res += dfs(pos-1, i, true, limit && i == a[pos]);
        else res += dfs(pos-1, i, sta, limit && i == a[pos]);
    }
    if(!limit) return dp[pos][pre][sta] = res;
    else return res;
}
ll solve(ll n)
{
    int pos = 0;
    while(n)
    {
        a[pos++] = n%10;
        n = n/10;
    }
    return dfs(pos-1, -1, false, true);
}
int main()
{
    int T;
    scanf("%d", &T);
    ll n;
    memset(dp, -1, sizeof(dp));
    while(T--)
    {
        cin >> n;
        cout << solve(n) << endl;
    }
    return 0;
}