bzoj1026 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 【输出样例二】 20Hint

 

【数据规模和约定】

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

 

#include <stdio.h>
#include <string.h>
#include <algorithm>
using namespace std;
typedef long long LL;
int bit[20];
LL dp[20][15], n, m;
LL dfs(int pos, int d, int pre, int flag, int limit) {
    if (pos == 0 ) return 1;
    if ( !flag && !limit && dp[pos][pre] != -1 ) return dp[pos][pre];
    int num = limit ? bit[pos] : 9;
    LL ans = 0;
    for (int i = 0 ; i <= num ; i++) {
        if (!flag && abs(pre - i) < 2) continue;
        ans += dfs(pos - 1, abs(i - pre), i, flag && i == 0, limit && i == num);
    }
    if (!limit && !flag) dp[pos][pre] = ans;
    return ans;
}
LL solve(LL x) {
    int len = 0;
    while(x) {
        bit[++len] = x % 10;
        x /= 10;
    }
    LL ans = 0;
    ans += dfs(len, 30, 0, 1, 1);
    return ans ;
}
int main() {
    int t;
    memset(dp, -1, sizeof(dp));
    scanf("%lld%lld", &n, &m);
    printf("%lld\n", solve(m) - solve(n - 1));
    return 0;
}