uestc 1307 windy数

Description

windy定义了一种windy数。
不含前导零且相邻两个数字之差至少为2的正整数被称为windy数。
windy想知道，在A和B之间，包括A和B，总共有多少个windy数？

Input

包含两个整数，A B。
满足 1 <= A <= B <= 2000000000 。

Output

包含一个整数：闭区间[A,B]上windy数的个数。

Sample Input

1 10

Sample Output

9

不错的数位dp!

#include <iostream>
#include <stdio.h>
#include <math.h>
#include <string.h>
using namespace std;
#define M 23
int dp[M][12],pri[M];
int get(int t,int i,int pre){if(t==11){if(i==0)return 11;else return i;}if(fabs(i-pre)>=2)return i;return -1;
}
int dfs(int pos,int pre,int t,int flag){if(t==-1)return 0;if(pos==0)return 1;if(!flag&&t>=0&&dp[pos][t]!=-1)return dp[pos][t];int u=flag?pri[pos]:9,ans=0;for(int i=0;i<=u;i++)ans+=dfs(pos-1,i,get(t,i,pre),flag&&i==u);return flag?ans:dp[pos][t]=ans;
}
int solve(int x){int cnt=0;while(x){pri[++cnt]=x%10;x=x/10;}return dfs(cnt,11,11,1);
}
int main()
{int tcase;int n,m;memset(dp,-1,sizeof(dp));while( scanf("%d%d",&m,&n)!=EOF){printf("%d\n",solve(n)-solve(m-1));}return 0;
}