【bzoj1026】[SCOI2009]windy数

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 。

代码

#include<cstdio>
#include<iostream>
#include<cstring>
#include<cmath>
#include<algorithm>
int f[15][10][2][2],a[15];
using namespace std;
int solve(int n)
{
    int len=0;
    while (n!=0)
    {
        len++;
        a[len]=n%10;
        n/=10;
    }
    memset(f,0,sizeof(f));
    for (int i=0;i<=a[len];i++) f[1][i][(i==a[len])][(i==0)]=1;
    for (int i=2;i<=len;i++)
    {
        for (int j=0;j<=9;j++)
        {
            for (int k=1;k<=9;k++)
            {
                if (abs(j-k)<2) continue;
                f[i][j][0][0]+=f[i-1][k][0][0];
                if (j<a[len-i+1]) f[i][j][0][0]+=f[i-1][k][1][0];
                if (j==a[len-i+1]) f[i][j][1][0]+=f[i-1][k][1][0];
            }
            f[i][j][0][(j==0)]+=f[i-1][0][0][1];
            if (j>=2)
            {
                f[i][j][0][0]+=f[i-1][0][0][0];
                if (j<a[len-i+1]) f[i][j][0][0]+=f[i-1][0][1][0];
                if (j==a[len-i+1]) f[i][j][1][0]+=f[i-1][0][1][0];
            }
        }
    }
    int sum=f[len][a[1]][1][0];
    for (int i=0;i<=9;i++)
        sum+=f[len][i][0][0];
    return sum;
}

int main()
{
    int l,r;
    cin>>l>>r;
    printf("%d",solve(r)-solve(l-1));
}