windy数——数位dp打卡

windy数

思路：
预处理第i位取j的windy数有多少（任意填，无限制）
然后考虑边界，以及还有前导0的问题，最高位从1开始枚举，其他数从0开始枚举，必是可以枚举的，如果到了最后一位，那就再加一。

代码

#include <iostream>
#include <cmath>
#include <vector>
using namespace std;
const int N = 15;
int f[N][N];

void init(){
    for(int i = 0;i<=9;i++) f[1][i] = 1;
    for(int i = 2;i<=N;i++){
        for(int j = 0;j<=9;j++){
            for(int k = 0 ;k<=9;k++){
                if(abs(k-j)>=2)
                    f[i][j]+=f[i-1][k];
            }
        }
    }


}

int dp(int n){
    int res = 0;
    if(!n) return 0;
    vector<int> nums;
    while(n){
        nums.push_back(n%10);
        n/=10;
    }
    int last = -2;
    for(int i = nums.size()-1;i>=0;i--){
        int x = nums[i];
        for(int j = (i==nums.size()-1);j<x;j++){
            if(abs(j-last)>=2)
                res += f[i+1][j];
        }
        if(abs(x-last)<2) break;
        last = x;
        if(!i) res++;
    }

    //特判有前导0的数字
    for(int i = 1;i<nums.size();i++) {
        for (int j = 1; j <= 9; j++)
            res += f[i][j];
    }

    return res;
}

int main() {
    init();
    int l,r;
    cin>>l>>r;
    cout<<dp(r)-dp(l-1);

    return 0;
}