uestc 1307 windy数 数位DP

题目链接：http://acm.uestc.edu.cn/problem.php?pid=1307

windy数的含义是：

不含前导零且相邻两个数字之差至少为2的正整数
题目想知道在A和B之间，包括A和B，总共有多少个windy数？

简单的数位DP

代码：

#include<iostream>
#include<cstdlib>
#include<cstdio>
#include<cstring>
using namespace std;
int dp[15][10];//dp[i][j]表示长度为i且最高位为j的windy数的个数
void init()
{
   memset(dp,0,sizeof(dp));
   for(int i=0;i<10;i++) dp[1][i]=1;
   for(int i=2;i<=10;i++)
      for(int j=0;j<10;j++)
      {
         for(int k=0;k<=j-2;k++)
            dp[i][j]+=dp[i-1][k];


         for(int k=j+2;k<10;k++)
            dp[i][j]+=dp[i-1][k];
      }
   
}
int bit[20];
int solve(int n)
{
   if(n==0) return 0;
   int len=0;
   while(n)
   {
       bit[++len]=n%10;
       n/=10;
   }
   bit[len+1]=-10;

   int ans=0;

   bool flag=1;
   for(int i=1;i<len;i++)//先计入长度小于len的
    for(int j=1;j<=9;j++)
            ans+=dp[i][j];
    for(int j=1;j<bit[len];j++)//再计入最高位
           ans+=dp[len][j];


    for(int i=len-1;i>=1;i--)
    {
       for(int j=0;j<bit[i];j++)
           if(abs(bit[i+1]-j)>=2)
                ans+=dp[i][j];
       if(abs(bit[i+1]-bit[i])<2)
           {
               flag=0;
               break;
           }
    }
    if(flag ) ans++;
    return ans;
}
int main()
{
   int a,b;
   init();
   while(scanf("%d%d",&a,&b)!=EOF)
   {
           cout<<solve(b)-solve(a-1)<<endl;

   }
   return 0;
}