区间内的数不包含某个数字的数的个数（数位dp）

最新推荐文章于 2022-01-27 01:14:51 发布

原创最新推荐文章于 2022-01-27 01:14:51 发布 · 1.5k 阅读

1 ·

本内容遵循CC 4.0 BY-SA版权协议

标签

#动态规划

动态规划专栏收录该内容

36 篇文章

订阅专栏

博客介绍了如何使用数位动态规划（DP）解决一个特定问题：在给定的数区间[n, m]内，找出所有不包含数字4且不连续出现62的整数个数。文章首先详细解释了数位DP的预处理过程，然后展示了如何根据预处理的DP数组计算指定区间的解。代码部分用C++实现，包括初始化DP数组和计算区间答案的函数。

题目：https://www.acwing.com/problem/content/1087/

题意：大概意思就是给一个区间[n,m]，问这个区间内的数，不包含4并且不包含62(连着)的数的个数。例如，62315,73418,88914都不行，61152就可以。
1≤n≤m≤10^9

题解：数位dp。
和之前的几道数位dp题思路完全一样，只是预处理的f数组不同，这里的f数组也比较好推。f[i][j]表示总共有i位，最高位为j且满足题目要求的所有的方案数，预处理dp的时候注意不要包含4，和62即可。

在这里插入图片描述

#include <bits/stdc++.h>
//#define int long long
#define pb push_back
#define pii pair<int, int>
#define mpr make_pair
#define ms(a, b) memset((a), (b), sizeof(a))
#define x first
#define y second
typedef long long ll;
typedef unsigned long long LL;
const int inf = 0x3f3f3f3f;
const ll INF = 0x3f3f3f3f3f3f3f3f;
using namespace std;
inline int read() {
    char ch = getchar();
    int s = 0, w = 1;
    while (ch < '0' || ch > '9') {
        if (ch == '-') w = -1;
        ch = getchar();
    }
    while (ch >= '0' && ch <= '9') {
        s = s * 10 + ch - '0', ch = getchar();
    }
    return s * w;
}
const int N = 10;

int f[N][N];
// f[i][j]表示总共有i位，最高位为j，并且满足题目要求的
// 所有的方案数
void init() {
	//首先预处理只有一位的情况
    for (int i = 0; i <= 9; i++)
        if (i != 4) f[1][i] = 1;
    
    for (int i = 2; i < N; i++) {//总共位数
        for (int j = 0; j <= 9; j++) {//枚举最高位
            if (j == 4) continue;
            for (int k = 0; k <= 9; k++) {//枚举次高位
                if (k == 4 || j == 6 && k == 2) continue;
                f[i][j] += f[i - 1][k];
            }
        }
    }
}
//下面过程就和之前几道数位dp题类似了
int dp(int n) {
    if (!n) return 1;
    vector<int> nums;
    while (n) nums.push_back(n % 10), n /= 10;
    int res = 0;
    int last = 0;
    for (int i = nums.size() - 1; i >= 0; i--) {
        int x = nums[i];
        for (int j = 0; j < x; j++) {
        	//注意跳出情况
            if (j == 4 || (last == 6 && j == 2)) continue;
            res += f[i + 1][j];
        }
        //发现原数上不满足题目限制，直接全部跳出即可
        if (x == 4 || (last == 6 && x == 2)) break;
        if (!i) res++;
        last = x;
    }
    return res;
}
signed main() {
    init();

    int l, r;
    while (cin >> l >> r, l || r) {
        cout << dp(r) - dp(l - 1) << endl;
    }
    return 0;
}