HDU 3555 Bomb 数位DP

题目大意：求1~n中含有‘49’的数字的个数。

题目思路：数位DP水题，看代码吧

#include<iostream>
#include<algorithm>
#include<cstring>
#include<vector>
#include<stdio.h>
#include<stdlib.h>
#include<queue>
#include<math.h>
#include<map>
#define INF 0x3f3f3f3f
#define MAX 1000005
#define Temp 1000000000

using namespace std;

//st==0 不含4
//st==1 含4
//st==2 含49

int num[MAX];
long long dp[50][50];

long long dfs(int pos,int st,int limit)
{
    if(pos<0)
        return st==2;
    if(!limit && dp[pos][st]!=-1)
        return dp[pos][st];
    long long ans=0;
    int len=limit?num[pos]:9;
    for(int i=0;i<=len;i++)
    {
        if(st==0)
        {
            if(i==4)
                ans+=dfs(pos-1,1,limit&&i==len);
            else
                ans+=dfs(pos-1,0,limit&&i==len);
        }
        else if(st==1)
        {
            if(i==9)
                ans+=dfs(pos-1,2,limit&&i==len);
            else if(i==4)
                ans+=dfs(pos-1,1,limit&&i==len);
            else
                ans+=dfs(pos-1,0,limit&&i==len);
        }
        else if(st==2)
        {
            ans+=dfs(pos-1,2,limit&&i==len);
        }
    }
    if(!limit)
        dp[pos][st]=ans;
    return ans;
}

long long solve(long long a)
{
    memset(dp,-1,sizeof(dp));
    int len=0;
    while(a)
    {
        num[len++]=a%10;
        a/=10;
    }
    long long ans=dfs(len-1,0,1);
    return ans;
}

int main()
{
    int T;
    long long n;
    scanf("%d",&T);
    while(T--)
    {
        scanf("%lld",&n);
        long long ans=solve(n);
        printf("%lld
",ans);
    }
    return 0;
}