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;
}