1083. Windy数

Windy 定义了一种 Windy 数：不含前导零且相邻两个数字之差至少为 2 的正整数被称为 Windy 数。

Windy 想知道，在 A 和 B 之间，包括 A 和 B，总共有多少个 Windy 数？

输入格式

共一行，包含两个整数 A 和 B。

输出格式

输出一个整数，表示答案。

数据范围

1≤A≤B≤2×109

输入样例1：

1 10

输出样例1：

9

输入样例2：

25 50

输出样例2：

20

思路：

/*
首先我们预处理的时候是包含前导0的，因为包含前导0的数可能被位数更多的首位不是0的数利用
最高位为0时的情况就包含了所有位数少于n的情况，但是不能直接算，因为计算时是包含前导0的，前导0会导致某些数的限制条件增多

上面提到了枚举的第i位是最高位，那么不能填0，这里解释一下，如果我们填0，那么答案就会加上f[i+1][0],，举这样一个例子，
对于数字13，他是满足windy数定义的，那么加上前导0之后的013就不会被f[3][0]加进去，原因就是abs(0-1)<2，这样就导致答案漏掉。
*/

代码：

#include <bits/stdc++.h>
using namespace std;
const int N = 15;
int f[N][N];// 有i位且最高位为j的符合条件的数

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(j - k) >= 2)
                    f[i][j] += f[i - 1][k];
            }
        }
    }
}
int dp(int n)
{
    if (!n)
        return 0; // 都不算0 方便
    vector<int> num;

    int res = 0;
    int last = -2;
    while (n)
    {
        num.push_back(n % 10);
        n /= 10;
    }

    for (int i = num.size() - 1; i >= 0; i--)
    {
        int x = num[i];
        for (int j = i == num.size() - 1; j < x; j++)
        {
            if (abs(j - last) >= 2) // 左分支
                res += f[i + 1][j];
        }
        if (abs(x - last) < 2) // 为右分支准备
            break;
        else
            last = x;
        if (!i)
            res++;
    }
    // 特殊处理有前导零的数
    for (int i = 1; i <= num.size() - 1; 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) << endl;

    return 0;
}