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