BZOJ 1026 windy数 数位DP模板题

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模板题，直接DP也可以，我贴了个数位DP板子。

#include <bits/stdc++.h>
using namespace std;
int dp[25][25], digit[20], len;
int dfs(int pos, int pre, int status, int limit){
    if(pos < 1) return 1;
    if(status && !limit && dp[pos][pre] != -1) return dp[pos][pre];
    int num = limit?digit[pos]:9;
    int ret = 0;
    if(status == 0){
        for(int i = 0; i <= num; i++){
            ret += dfs(pos - 1, i, i, limit&&(i == num));
        }
    }
    else{
        for(int i = 0; i <= num; i++){
            if((i - pre)*(i - pre) < 4) continue;
            ret += dfs(pos - 1, i, 1, limit && (i == num));
        }
    }
    if(status && !limit){
        dp[pos][pre] = ret;
    }
    return ret;
}
int f(int num){
    len = 0;
    while(num){
        digit[++len] = num%10;
        num /= 10;
    }
    return dfs(len, 0, 0, 1);
}
int main(){
    int A, B;
    memset(dp, -1, sizeof(dp));
    scanf("%d%d", &A, &B);
    int ans = f(B) - f(A - 1);
    cout << ans << endl;
}