hdu 3652(数位dp 记忆化搜索方法)

题目链接：点击打开链接

//数位dp   记忆化搜索方法

#include
#include

int digit[20];//存各位上的数字
int dp[20][3][13];//数位  前缀标记 余数

int l;

int dfs(int pre,int pos,int mod,int doing)
{
    if(pos==-1)//搜到结尾
    {
        if(mod==0&&pre==2) return 1;//余数为零 并且包含13  返回1说明这种状态合法结果需要加1
        return 0;
    }
    if(!doing&&dp[pos][pre][mod]!=-1) return dp[pos][pre][mod];//以前标记过的状态直接返回

    int i,End,ans;
    End=doing?digit[pos]:9;//判断枚举上界 
    ans=0;
    for(i=0;i<=End;i++)
    {
        if(pre==2)//如果前缀包含13  枚举后面的位时前缀一定包含13
            ans+=dfs(2,pos-1,(mod*10+i)%13,doing && i==End);
        else if(pre==1&&i==3)//如果前一位是1 当前位时3 则枚举以后位时前缀包含13
        {
            ans+=dfs(2,pos-1,(mod*10+i)%13,doing && i==End);
        }
        else if(i==1)//如果当前位是1 则枚举下一位时  前一位是1
        {
            ans+=dfs(1,pos-1,(mod*10+i)%13,doing && i==End);
        }
        else// 其他情况为前缀不包含13 且前一位不为1 的情况
        {
            ans+=dfs(0,pos-1,(mod*10+i)%13,doing && i==End);
        }
    }

    if(!doing)
        dp[pos][pre][mod]=ans;
    return ans;
}
int solve(int n)
{

    l=0;
    memset(digit,0,sizeof(digit));
    memset(dp,-1,sizeof(dp));


    while(n>0)//把n的各位上的数存到digit中
    {
    digit[l++]=n%10;
    n=n/10;
    }
    return dfs(0,l-1,0,1);
}

int main()
{
    int n;
    while(scanf("%d",&n)!=EOF)
    {
        printf("%d\n",solve(n));
    }
    return 0;
}