hdu 3555 Bomb 【数位DP】

题目链接:http://acm.hdu.edu.cn/showproblem.php?pid=3555

题意:上一题是不要62 这个是“不要49”

代码:

#include <stdio.h>
#include <ctime>
#include <math.h>
#include <limits.h>
#include <complex>
#include <string>
#include <functional>
#include <iterator>
#include <algorithm>
#include <vector>
#include <stack>
#include <queue>
#include <set>
#include <map>
#include <list>
#include <bitset>
#include <sstream>
#include <iomanip>
#include <fstream>
#include <iostream>
#include <ctime>
#include <cmath>
#include <cstring>
#include <cstdio>
#include <time.h>
#include <ctype.h>
#include <string.h>
#include <assert.h>

using namespace std;

#define N 50

using namespace std;
int bit[N];
long long dp[N][3];

/*
      dp[i][0]:前i位不含 49 的个数。

    dp[i][1]:前i位不含 49 数且i+1位是4的个数。

    dp[i][2]:前i位含 49 的个数。
*/

long long dfs(int pos, int st, bool flag) 
{
    if (pos == 0) return st == 2;
    if (flag && dp[pos][st] != -1) return dp[pos][st];
    long long ans = 0;
    int u = flag ? 9 : bit[pos]; 

    for (int d = 0;d <= u;d++) 
    {
        if (st == 2 || (st == 1 && d == 9))
            ans += dfs(pos - 1, 2, flag || d<u);
        else if (d == 4)
            ans += dfs(pos - 1, 1, flag || d<u);
        else 
            ans += dfs(pos - 1, 0, flag || d<u);
    }
    if (flag) dp[pos][st] = ans;
    return ans;
}

long long solve(long long n) 
{
    int len = 0;
    while (n) 
    {
        bit[++len] = n % 10;
        n /= 10;
    }
    return dfs(len, 0, 0);
}

int main() 
{
    long long n;
    int t;
    scanf("%d",&t);
    while (t--) 
    {
        scanf("%lld", &n);
        memset(dp, -1, sizeof(dp));
        printf("%lld\n", solve(n));
    }
    return 0;
}
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值