P2657 [SCOI2009] windy数--数位dp,dfs

题目
方法一：推出状态转移方程直接用for循环递推

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const ll N = 1e3 + 5;
const ll MOD = 1e9 + 7;
const ll INF = 0x7fffffff;
ll f[N][10];//表示有i位数字且最高位是j的情况下满足条件的总和
void init() {//预处理f数组
    for (ll i = 0; i <= 9; i++) {//只有1位的情况下
        f[1][i] = 1;
    }
    //两位数及以上的情况
    for (ll i = 2; i < 18; i++) {

        for (ll j = 0; j <= 9; j++) {//枚举最高位数字
            for (ll k = 0; k <= 9; k++) {//枚举下一位
                if(abs(j-k)>=2)
                f[i][j] += f[i - 1][k];
            }
        }

    }
}

ll dp(ll n) {//从0到n中有多少数满足要求
    if (!n) return 0;//边界，n为0的情况

    vector<ll> nums;
    while (n) nums.push_back(n % 10), n /= 10;//把数的每一位都存下来

    ll res = 0;//答案
    ll last = -2;//保存上一位的信息
    for (ll i = nums.size() - 1; i >= 0; i--) {
        ll x = nums[i];
        for (ll j = i == nums.size() - 1; j < x; j++)
            if (abs(j - last) >= 2)
                res += f[i + 1][j];
        if (abs(x - last) >= 2) last = x;//对下次来说是上一位的信息
        else break;
        if (!i) res++;
    }
    // 特殊处理有前导零的数
    for (int i = 1; i < nums.size(); i++)
        for (int j = 1; j <= 9; j++)
            res += f[i][j];

    return res;
}

int main() {
    ios_base::sync_with_stdio(false), cin.tie(0), cout.tie(0);
    init();
    ll l, r;
    cin >> l >> r;
    cout << dp(r) - dp(l - 1) << "\n";

    return 0;
}

方法二：dfs递归求解

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const ll N = 15;
const ll MOD = 1e9 + 7;
const ll INF = 0x7fffffff;
ll dp[N][N];
ll nums[N];//保存数位
ll dfs(ll len, ll pre, bool ismax, bool iszero) {//len为长度，pre记录上一位的状态，ismax表示上一位数是否取到了为最高位
    if (!len)return 1;//递归边界，长度为0时返回1
    if (dp[len][pre] != -1 && !ismax && !iszero)return dp[len][pre];//如果找过了,且不是最高位,且无前导零直接返回结果
    ll top = ismax ? nums[len] : 9;//当前最高位是多少，例如n=655，假如最高位是6，第二位最大只能是5=num[len]，否则最大可以是 9。
    ll ans = 0;
    for (ll i = 0; i <= top; i++) {
        if (abs(pre - i) < 2)continue;
        if (iszero&&i==0)ans += dfs(len - 1, -2, ismax && i == top, true);
        else ans += dfs(len - 1, i, ismax && i == top, false);
    }
    if (!ismax && !iszero)dp[len][pre] = ans;//记忆化
    return ans;
}
ll solve(ll n) {
    ll len = 0;
    while (n)nums[++len] = n % 10, n /= 10;//取出每一位
    memset(dp, -1, sizeof(dp));//初始化-1（因为有可能某些情况下的方案数是0）
    return dfs(len, -2, true, true);
}
int main() {
    ios_base::sync_with_stdio(false), cin.tie(0), cout.tie(0);

    ll l, r;
    cin >> l >> r;
    cout << solve(r) - solve(l - 1) << "\n";

    return 0;
}