Bomb [数位DP]

题目与传统数位DP略有不同,是求区间内带有49的数字的个数。
如果是求不带有49的,那会非常简单,如果求带有49的,只需要在每次dfs找到可行情况的时候将后面的结果都加上。
如果上一位是4,且这一位是9,那么根据是否是limit限制,确定之后有的个数加上。 否则直接dfs下一位。

#include <cstdio>
#include <cstring>
#include <iostream>
#include <cmath>
using namespace std;
typedef long long LL;
const int N = 30;
LL z[N] = {1};

int t;
int bit[N];
LL dp[N][2], n;
LL dfs(int len, bool is4, bool ismax) {
    if (len == 0)
        return 0;
    if (!ismax && dp[len][is4] >= 0)
        return dp[len][is4];

    LL cnt = 0;
    int mmax = ismax ? bit[len] : 9;
    for (int i = 0; i <= mmax; i++) {
        if (is4 && i == 9) {
            cnt += ismax ? n % z[len-1] + 1 : z[len-1];
        } else {
            cnt += dfs(len - 1, i == 4, ismax && bit[len] == i);
        }
    }
    return ismax ? cnt : dp[len][is4] = cnt;
}
LL solve(LL n) {
    int len = 0;
    while (n) {
        bit[++len] = n % 10;
        n /= 10;
    }
    bit[len + 1] = 0;
    return dfs(len, false, true);
}

int main()
{
    memset(dp, -1, sizeof(dp));
    for (int i = 1; i < N; i++) {
        z[i] = z[i - 1] * 10;
    }
    scanf("%d", &t);
    while (t--) {
        scanf("%I64d", &n);
        printf("%I64d\n", solve(n));
    }
    return 0;
}
  • 0
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值