AcWing 1083. Windy数 数位dp例题

此题中，dp数组保存存在前导0的windy数（因为在使用时，当前位的前面会存在非0数）

结果为不存在前导0的windy数（循环过程中第一次不包含0）。

#include<bits/stdc++.h>
using namespace std;
typedef long long LL;

LL dp[15][15];//i起头，长度为j
LL sum[15];
LL DP(int num, int len)
{
    return dp[num][len];
}
LL Count(LL num)
{
    string nums = to_string(num);
    LL res = 0;
    int prev = -10;
    for (int i = 0; i < nums.size(); i++)
    {
        int num = nums[i] - '0';
        

        for (int j = (i==0?1:0); j < num; j++)
        {
            if (abs(j - prev) >= 2)
                res += DP(j, nums.size() - i);
        }
        if (abs(num - prev) < 2)
            break;
        if (i == nums.size() - 1)
            res++;
        prev = num;
    }
    for (int i = nums.size() - 1; i >= 1; i--) res += sum[i];
    return res;
}
void YD()
{
    LL a, b;
    for (int i = 0; i < 10; i++) {
        dp[i][1] = 1;
    }

    for (int j = 2; j <= 11; j++)
    {
        for (int i = 9; i >= 0; i--)
        {
            for (int k = 0; k < 10; k++)
            {
                if (abs(i - k) >= 2)
                    dp[i][j] += dp[k][j - 1];
            }
        }
    }

    for (int i = 0; i <= 11; i++)
    {
        for(int j=1;j<10;j++)
            sum[i] += dp[j][i];
    }
    sum[1] += dp[0][1];

    cin >> a >> b;

    LL res = Count(b);
    res -= Count(a - 1);
    cout << res << endl;



}

int main()
{
    ios_base::sync_with_stdio(false);
    cin.tie(nullptr);
    cout.tie(nullptr);
    int T = 1;
    //cin >> T;
    while (T--)
    {
        YD();
    }
    return 0;
}