BZOJ1026 windy数 数位DP

题目链接：http://www.lydsy.com/JudgeOnline/problem.php?id=1026

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

分析：数位DP，定义dp(i,j)为高一位为j的i位数中满足题意的数的个数。

实现代码如下：

#include <cstdio>#include <cstring>using namespace std;int dp[25][25];int dig[25],len;int dfs(int pos,int pre,int z,int lim){    if(pos<=0) return 1;    if(z&&!lim&&dp[pos][pre]!=-1) return dp[pos][pre];    int num=lim?dig[pos]:9;    int ans=0;    if(z==0)    {        for(int i=0;i<=num;i++)          ans+=dfs(pos-1,i,i,lim&&(i==num));    }    else    {        for(int i=0;i<=num;i++)        {            if((i-pre)*(i-pre)<4) continue;            ans+=dfs(pos-1,i,1,lim&&(i==num));        }    }    if(z&&!lim) dp[pos][pre]=ans;    return ans;}int count(int x){    len=0;    while(x)    {        dig[++len]=x%10;        x/=10;    }    memset(dp,-1,sizeof(dp));    return dfs(len,0,0,1);}int main(){    int l,r;    while(scanf("%d%d",&l,&r)!=-1)    {        printf("%d\n",count(r)-count(l-1));    }    return 0;}