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 ;

}