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