hdu3555（数位dp记忆化搜索）

记忆化搜索解法

题目链接：点击打开链接

#include<stdio.h>
#include<string.h>

__int64 dp[25][3];//dp[pos][pre] pos记录位数 pre记录前缀状态0表示前缀不包含49且pos+1不是4 1表示前缀不包含49且pos+1位是4 2表示前缀包含49
int digit[25];//存数的各位上的数字
int l;//存数的位数

__int64 dfs(int pos,int pre,int doing)
{
    if(pos==-1)
    {
        if(pre==2)//如果当前枚举的数含有49 返回1
            return 1;
        return 0;
    }
    if(!doing&&dp[pos][pre]!=-1)return dp[pos][pre];//如果枚举的是自由位可以用以前记录过的答案
    int End,i;
    End=doing?digit[pos]:9;//根据枚举的是不是自由位来决定枚举范围

    __int64 ans=0;

    for(i=0;i<=End;i++)
    {
        if(pre==2)//前缀包含49则后面的位不论是什么这个数的都包含49 pre=2不会改变
        {
            ans+=dfs(pos-1,2,doing &&i==End);
        }
        else if(i==9&&pre==1)//如果pos+1位为4当前位为9 则以后的数位以此为前缀包含49 pre=2 
        {
            ans+=dfs(pos-1,2,doing &&i==End);
        }
        else if(i==4)//如果pre！=2 且当前位为4 则以后的数位pre=1
        {
            ans+=dfs(pos-1,1,doing &&i==End);
        }
        else// 以上条件都不满足的话pre=0
            ans+=dfs(pos-1,0,doing &&i==End);
    }

    if(!doing)
        dp[pos][pre]=ans;
    return ans;
}
__int64 solve(__int64 x)
{
        int i;
        l=0;
        memset(digit,0,sizeof(digit));
        memset(dp,-1,sizeof(dp));

        while(x>0)
        {
            digit[l++]=x%10;
            x=x/10;
        }
        return dfs(l-1,0,1);
}

int main()
{
    int m;
    scanf("%d",&m);
    __int64 n;
    while(m--)
    {
        scanf("%I64d",&n);
        printf("%I64d\n",solve(n));

    }
    return 0;
}