【BZOJ1026/SCOI2009】windy数

                               1026: [SCOI2009]windy数

                                              Time Limit: 1 Sec  Memory Limit: 162 MB
                                                       Submit: 10470  Solved: 4899

Description

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

Input

　　包含两个整数，A B。

Output

　　一个整数

Sample Input

【输入样例一】
1 10
【输入样例二】
25 50

Sample Output

【输出样例一】
9
【输出样例二】
20

HINT

【数据规模和约定】

100%的数据，满足 1 <= A <= B <= 2000000000 。

 

解析：

       数位DP。

       注意处理前导0的情况。

 

代码：
 

#include <bits/stdc++.h>
using namespace std;
 
const int Max=12;
int l,r,n;
int f[Max][Max][2][2];
char ch[Max];
 
inline int dfs(int pos,int pre,int limit,int tag)
{
    if(pos==n+1) return 1;
    if(~f[pos][pre][limit][tag]) return f[pos][pre][limit][tag];
    int mx=limit?ch[pos]-'0':9,ans=0,now=tag;
    for(int i=0;i<=mx;i++)
    {
      if(abs(i-pre)<2&&now) continue;
      if(!now&&i) now=1;
      ans+=dfs(pos+1,i,limit&(i==mx),now);
    }
    return f[pos][pre][limit][tag]=ans;
}
 
inline int solve(int num)
{
    memset(f,-1,sizeof(f));
    sprintf(ch+1,"%d",num);
    n=strlen(ch+1);
    return dfs(1,-2,1,0);
}
 
int main()
{
    cin>>l>>r;
    cout<<solve(r)-solve(l-1);
    return 0;
}