HYSBZ 1026 windy数 (数位DP)

数位DP第一发。发现数位DP与我自己YY的数位DP不太一样啊。。。sad。。

数位DP是用dfs+记忆化搜索，从后面往前推。第一次写数位DP，调试了好长时间。。。

代码如下：

#include <iostream>
#include <string.h>
#include <math.h>
#include <queue>
#include <algorithm>
#include <stdlib.h>
#include <map>
#include <set>
#include <stdio.h>
using namespace std;
#define LL long long
#define pi acos(-1.0)
const int mod=100000000;
const int INF=0x3f3f3f3f;
const double eqs=1e-8;
LL dp[11][12], c[20];
LL dfs(int cnt, int pre, int maxd, int zero)
{
        int i, r;
        if(cnt==-1) return 1;
        if(maxd&&dp[cnt][pre]!=-1&&zero) return dp[cnt][pre];
        LL ans=0;
        if(maxd) r=9;
        else r=c[cnt];
        for(i=0; i<=r; i++) {
                if(!zero||abs(i-pre)>=2) {
                        ans+=dfs(cnt-1,i,maxd||i<r,zero||i);
                }
        }
        if(maxd&&zero) dp[cnt][pre]=ans;
        return ans;
}
LL Cal(LL x)
{
        int i;
        int cnt=0;
        while(x) {
                c[cnt++]=x%10;
                x/=10;
        }
        return dfs(cnt-1, 11, 0, 0);
}
int main()
{
        LL a, b, ans1, ans2, i;
        memset(dp,-1,sizeof(dp));
        while(scanf("%lld%lld",&a,&b)!=EOF) {
                ans1=Cal(a-1);
                ans2=Cal(b);
                printf("%lld\n",ans2-ans1);
        }
        return 0;
}