hdu 3555 数位dp水题 记忆化搜索做法

#include<iostream>
#include<cstring>
#include<cstdio>
using namespace std ;
const int maxn = 20;
__int64 dp[maxn][3] ;//dp[i][flag] ,flag = 2，表示已经有49，flag == 1，表示没有49，这一位是4，
int bit[maxn] ;    //flag == 0, 什么都没有
__int64 dfs(int pos , int flag , int lim)
{
    if(pos == 0)
    return (flag == 2);
    if(!lim && dp[pos][flag] != -1)return dp[pos][flag] ;
    int num = lim ? bit[pos] : 9 ;
    __int64 ans  = 0;
    for(int i = 0; i <= num ;i++)
    {
        int flag_x = flag ;
        if(flag == 1 && i == 9)
        flag_x = 2;
        else if(flag == 0 && i == 4)
        flag_x = 1;
        else if(flag == 1 && i!=4)
        flag_x = 0;
        ans+=dfs(pos - 1 , flag_x ,lim && (i==num)) ;
    }
    if(!lim)dp[pos][flag] = ans;
    return ans ;
}
int main()
{
    int T ;
    scanf("%d" ,&T) ;
    while(T--)
    {
        __int64 n ;
        scanf("%I64d" ,&n) ;
        __int64 t = n ; int len = 0;
        while(t)
        {
            bit[++len] = t%10 ;
            t/=10 ;
        }
        memset(dp , -1 ,sizeof(dp)) ;
        printf("%I64d\n" ,dfs(len , 0 , 1)) ;
    }
    return 0 ;
}