Bomb HDU - 3555 数位dp

题目链接：https://vjudge.net/problem/HDU-3555
题意：求出现49的数的个数
思路：用sta区分前一位是否是4。先算不满足条件的数然后减去即可。
反向求

#include <cstdio>
#include <iostream>
#include <algorithm>
#include <cstring>
#include <cmath>
#include <vector>
#include <queue>
#include <stack>
using namespace std;
long long a[50],dp[50][2];
long long dfs(long long pos,long long pre,long long sta,bool limit)
{
    if(pos==-1)
    {
            return 1;
    }
    if(!limit&&dp[pos][sta]!=-1)
        return dp[pos][sta];
    long long up=limit?a[pos]:9;
    long long tmp=0;
    for(long long i=0; i<=up; i++)
    {
        if(pre==4&&i==9)
           continue;
        tmp+=dfs(pos-1,i,i==4,limit&&i==a[pos]);
    }
    if(!limit)
    {
         dp[pos][sta]=tmp;
    }
    return tmp;
}
long long solve(long long x)
{
    long long pos=0;
    while(x)
    {
        a[pos++]=x%10;
        x/=10;
    }
    return dfs(pos-1,-1,0,true);
}
int main()
{
    int T;
    scanf("%d",&T);
    while(T--)
    {
        memset(dp,-1,sizeof(dp));
        long long n;
        scanf("%lld",&n);
        printf("%lld\n",(n-solve(n)+1));
    }
    return 0;
}

正向求
转自：https://www.bbsmax.com/A/1O5Eg1Ybz7/

#include<cstdio>
#include<iostream>
#include<algorithm>
#include<cstring>
#include<sstream>
#include<cmath>
#include<stack>
#include<cstdlib>
#include <vector>
#include<queue>
using namespace std;

using namespace std;
#define ms(a,b) memset(a,b,sizeof(a))
#define lson rt*2,l,(l+r)/2
#define rson rt*2+1,(l+r)/2+1,r
typedef unsigned long long ull;
typedef long long ll;
const int MAXN=1e4+5;
const double EPS=1e-8;
const int INF=0x3f3f3f3f;
const int MOD = 1e9+7;

#define ll long long
#define llu unsigned long long
#define INF 0x3f3f3f3f
#define PI acos(-1.0)
const int maxn =  1e5+5;
const int  mod = 1e9+7;

int bit[40];
ll f[40][4];
ll dp(int pos,int st,bool flag)
{
    //pos为位，st为状态，st=0：没有49，st=1：前一位为4，st=2，表示49都已经出现过了
    //flag表示高位与原数是否相同
    if(pos == 0)
        return st == 2;
    if(flag && f[pos][st] != -1)
        return f[pos][st];
    ll ans = 0;
    int x;
    if(flag)
        x = 9;
    else
        x = bit[pos];
    //cout<<x<<endl;
    for(int i = 0; i <= x; i++)
    {
        if((st == 2) || (st == 1 && i == 9))    //如果上一位已经有49（st==2）或者上一位为4（st==1）当前位为9
            ans += dp(pos-1,2,flag || i<x);     //只能加上上一个状态，且上一个状态一定为已经有49了
        else if(i == 4)
            ans += dp(pos-1,1,flag || i<x);     //当前为4和前一位也为4的状态是一样的
        else                                    //维持st=0的状态
            ans += dp(pos-1,0,flag || i<x);
    }
    if(flag)
        f[pos][st] = ans;   //记忆化
    return ans;
}

ll calc(ll x)
{
    int len = 0;
    while(x)
    {
        bit[++len] = x % 10;
        x /= 10;
    }
    //cout<<len<<endl;
    return dp(len,0,0);
}
int main()
{
    int t;
    scanf("%d",&t);
    memset(f,-1,sizeof f);
    while(t--)
    {
        ll n;
        scanf("%lld",&n);
        printf("%lld\n",calc(n));
    }
}