AcWing 1083. Windy数

题目传送门
视频讲解

#include <bits/stdc++.h>

using namespace std;

const int N = 32;

int l, r;
int a[N],al;
int f[N][N];

int dp(int pos, int st, int op) {
    if (!pos) return 1;
    if (!op && ~f[pos][st]) return f[pos][st];
    int res = 0, up = op ? a[pos] : 9;
    for (int i = 0; i <= up; i++)
        if (abs(st - i) >= 2) //分拆任务+合并结果
            //如果st=前面全是0，!i:本位还是0，那么前序值继续-2
            res += dp(pos - 1, !i && st == -2 ? -2 : i, op && i == a[pos]);
    return op ? res : f[pos][st] = res;
}

int calc(int x) {
    memset(f, -1, sizeof f);
    al = 0;
    while (x) a[++al] = x % 10, x /= 10;
    //如果前面全是0，那么st=-2,否则就是前序的数字
    return dp(al, -2, 1);
}

int main() {
    cin >> l >> r;
    cout << calc(r) - calc(l - 1) << endl;
    return 0;
}