Windy数（数位dp）

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。

源代码：

#include <iostream>
#include <cstring>
#include <algorithm>
#include <cmath>
#include <stdio.h>
using namespace std;
int dp[20][10];
int a[20];
int digit(long long x){
    memset(a,0,sizeof(a));
    int length = 0;
    while (x){
        a[++length] = x % 10;
      //  cout << a[length];
        x /= 10;
    }
    //cout << endl;
    return length;
}
long long solve (long long x){
    long long ans = 0;
    int length = digit(x);
    //for (int i = 1; i <= length; i++)
      //  cout << a[i];
    //cout << endl;
    for (int i = 1; i <= a[length]-1; i++)
        ans += dp[length][i];
    for (int i = length - 1; i > 0; i--)
        for (int j = 1; j <= 9; j++)
            ans += dp[i][j];
    for (int i = length - 1; i > 0; i--){
        for (int j = 0; j <= a[i]-1; j++)
            if (abs(a[i+1]-j)>=2)
                ans += dp[i][j];
        if (abs(a[i+1] - a[i]) < 2)
            break;
    }
    return ans;
}
int main(){
    memset(dp,0,sizeof(dp));
    for (int i = 0; i <= 9; i++)
        dp[1][i] = 1;
    for (int i = 2; i <= 10; i++)
        for (int j = 0; j <= 9; j++)
            for (int k = 0; k <= 9; k++)
                if (abs(j - k) >= 2)
                    dp[i][j] += dp[i-1][k];
    long long a,b;
    scanf("%lld%lld",&a,&b);
    printf("%lld\n",solve(b+1)-solve(a));
    return 0;
}