HDU 2089 不要62--数位dp,dfs

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

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

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

    }
}

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

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

    ll res = 0;//答案
    ll last = 0;//保存上一位的信息
    for (ll i = nums.size() - 1; i >= 0; i--) {
        ll x = nums[i];
        for (ll j = 0; j < x; j++) {//枚举最高位
            if (j == 4 || last == 6 && j == 2) continue;
            res += f[i + 1][j];//从0到i有i+1种情况，最高位是j是
        }
        if (x == 4 || last == 6 && x == 2) break;
        last = x;//对下次来说是上一位的信息
        if (!i) res++;
    }
    return res;
}

int main(){
    ios_base::sync_with_stdio(false), cin.tie(0),cout.tie(0);
    init();
    ll l, r;
    while (cin >> l >> r, 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][10];//dp[len][i]表示不含4和62的前提下，剩余长度为len且首位为i的个数
ll nums[N];//保存数位
ll dfs(ll len, ll pre, bool ismax) {//len为长度，pre记录上一位的状态，ismax表示上一位数是否取到了为最高位
    if (!len)return 1;//递归边界，长度为0时返回1
    if (dp[len][pre] != -1 && !ismax)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 (i == 4 || (i == 2 && pre == 6))continue;//不满足题意的情况
        ans += dfs(len - 1, i, ismax && i == top);//递归下去
    }
    if (!ismax)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, 0, true);
}
int main() {
    ios_base::sync_with_stdio(false), cin.tie(0), cout.tie(0);

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

    return 0;
}