【数位DP】不要62 HDU - 2089 入门理解

DP - 数位DP

不要62 HDU - 2089

解题思路

对于数 $x$，先用 dfs 进行数位 dp 计算 $[1,x]$ 中满足条件的个数，之后利用前缀和思想求解 $m$ 和 $n$ 的方案数之差。

变量解释：

• $dp[i][j]$ ：进行到第 $i$ 位数时，当前数为 $6$ 或者不为 $6$ 的方案数。

• $len$ ：当前递归到第几位。

• $pre$ ：前一位数字。用于排除 $62$ 的情况。

• $sta$ ：当前数字是否是 $6$。用于记忆化搜索。

• $limit$ ：记录当前是否有约束条件。例如，$n=345$ 时，若当前在处理第三位，且数字为 $3$，那么递归下去的第二位就只能从 $[0,4]$ 中取。若第二位取 $[0,3]$，第三位任意选择，都不会超过上界；但若第二位取 $4$，那么第三位就有了约束，只能从 $[0,5]$ 中取。

  由此可见，判断递归的下一层是否需要约束，是由两个因素决定的：

  1. 是否前面一路都是约束的状态，传递过来的（也就是当前在处理最大值的各个位，如最大值是 $345$，那么就是 $3xx$，$34x$ 这样传递过来）。
  2. 当前的位数是否是最大值某一位的最值。

  在代码中体现如下：

  tmp += dfs(len - 1, i, i == 6, limit && i == up);  //若现在是最大值的某一位，且该位又达到了最值，那么接下来的位也要有约束
  

参考代码

#include<stdio.h>
#include<iostream>
#include<vector>
#include<cstring>
#include<cstdio>
#include<climits>
#include<cmath>
#include<algorithm>
#include<queue>
#include<deque>
#include<map>
#include<set>
#include<stack>
#define LOCAL  //提交时一定注释
#define VI vector<int>
using namespace std;
typedef long long LL;
typedef double db;
const int inf = 0x3f3f3f3f;
const LL INF = 1e18;
const int N = 20;

int readint() {int x; scanf("%d", &x); return x;}

int a[N], dp[N][2];
LL ans = 0;
void init(){
    memset(dp, -1, sizeof(dp));
}

int dfs(int len, int pre, int sta, bool limit) {
    if (len == 0) return 1;  //成功将所有的位都递归完，说明当前这个方案可行，方案数+1
    if (!limit && dp[len][sta] != -1) return dp[len][sta];  //若之前已经记录过，直接返回值
    int up = limit ? a[len] : 9;  //若当前进行到最大值的各个位，那么要有对应的约束条件
    int tmp = 0;  //记录当前的结果
    for(int i = 0; i <= up; i++) {
        if (i == 4) continue;  //题目条件，不考虑有4的情况
        if (pre == 6 && i == 2) continue;  //不考虑前面是6，现在是2的情况
        tmp += dfs(len - 1, i, i == 6, limit && i == up);  //若现在是最大值的某一位，且该位又达到了最值，那么接下来的位也要有约束
    }
    if (!limit) dp[len][sta] = tmp;  //若没有约束，则可以记录，之后相同状态的直接沿用
    return tmp;
}

int solve(int n) {
    int len = 0;
    while (n) {
        a[++len] = n % 10;  //记录每一位的最大值
        n /= 10;
    }
    return dfs(len, 0, 0, true);
}
int main() {
#ifdef LOCAL
    freopen("input.txt", "r", stdin);
//   freopen("output.txt", "w", stdout);
#endif
    int n, m;
    while (~scanf("%d%d", &n, &m), n) {
        init();
        printf("%d\n", solve(m) - solve(n - 1));  //前缀和思想
    }
    return 0;
}