数位DP入门篇之HDU2089——不要62

记忆化搜索

#include<iostream>
#include<cstring>
#include<cstdio>
using namespace std;
int n,m;
int dp[20][3];
int num[20],pos;
int dfs(int pos,int s,int inf)
{
    if(s==2) return 0;
    if(pos==-1) return s==0||s==1;
    if(!inf&&dp[pos][s]!=-1) return dp[pos][s];
    int end=inf?num[pos]:9;
    int sum=0;
    for(int i=0;i<=end;i++){
        int have=s;
        if(i==4) have=2; //把含4的状态去掉，就变成判断是否含62了 ，这样子 就跟HDU3555一样的操作方式
        else{
            if(s==0&&i==6) have=1;
            if(s==1&&i==2) have=2;
            if(s==1&&i!=2&&i!=6) have=0;
        }
        sum+=dfs(pos-1,have,inf&&i==end);
    }
    return inf?sum:dp[pos][s]=sum;
}
void Init(int x)
{
    pos=0;
    while(x){
        num[pos++]=x%10;
        x/=10;
    }
    memset(dp,-1,sizeof dp);
}
int Cal(int x)
{
    Init(x);
    return dfs(pos-1,0,1);
}
int main()
{
    while(scanf("%d%d",&n,&m)&&(n||m)){
        printf("%d\n",Cal(m)-Cal(n-1));
    }
    return 0;
}