hdu 3555 bomb 数位dp

题目链接

和上一题差不多， 可以用总数-不含49的个数。

#include<bits/stdc++.h>
using namespace std;
#define mem1(a) memset(a, -1, sizeof(a))
#define ll long long
ll dp[20][20], digit[20];
int len;
ll dfs(int len, bool state, bool fp) {     //state表示这一位是否有限制
    if(!len)                                //fp表示len这一位是否只能取到小于等于digit[len]的数
        return 1;
    if(!fp && dp[len][state]!=-1)
        return dp[len][state];
    ll ret = 0, maxx = fp?digit[len]:9;
    for(int i = 0; i<=maxx; i++) {
        if(i==9&&state)
            continue;
        ret += dfs(len-1, i==4, fp&&i == maxx); //对于这一题， 当这一位为6的时候， 下一位就会有限制
    }
    if(!fp)
        return dp[len][state] = ret;
    return ret;
}
ll cal(ll n) {
    len = 0;
    ll tmp = n;
    while(n) {
        digit[++len] = n%10;
        n/=10;
    }
    return tmp-dfs(len, false, true);
}
int main()
{
    int t;
    ll b;
    cin>>t;
    while(t--) {
        scanf("%I64d", &b);
        mem1(dp);
        printf("%I64d
", cal(b)+1);
    }
}