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