数位dp优化（2022杭电多校7-1006）

数位dp优化（2022杭电多校7-1006）

题目大意

求 ${\sum }_{i=l}^r$ $f^k(i,b,d)$ ，$f(i,b,d)$表示用b进制表示i，数位d数位的次数

数据范围： $1\le b\le 10^9$, $0\le d<b$ ,$0\le k\le 10^9$, $1\le l\le r\le 10^{18}$

并且规定 $0^0$ = $0$

思路

考虑数位dp，但是这题每个数位的种类太多，不能像以前数位dp一样用循环累加的方式处理，所以我们只能将循环中会出现的几种情况进行分类讨论，然后再分组求和：

1、当目前枚举到了数位上界时：按照数位上界的做法继续dfs，如果上界为0，则退出递归以免后面特殊处理0位时重复计数。

2、当枚举到第0个结果时：判断当前是否为前导0，如果为前导0，则不能将这个0计入最后的结果，如果不是前导0，并且d=0时，这个0将计入最后的结果中。

3、当我们枚举到d时：这里要进行分类讨论，如果d=0，那么d已经统计过了，不再重复计数，如果d不为0，则需要0将d单独讨论计数。

4、一般情况：算出除了上述情况以外的其他情况的个数，继续进行dfs，将得到的结果乘上一般情况的个数即为最后答案。

其他的条件与一般的计数类数位dp一致。

注意：要判断前导0的情况，前导0不能作为dp记录方案数，要单独计算。

关于k次方的处理：本题需要处理最后的结果为答案的k次方，但是我们发现，当我们枚举到最后一位后，我们对这个数统计的个数已经确定，只需要将这个结果进行快速幂即可得到当前这个数的答案，然后将该结果返回上一级即可。

代码

#pragma comment(linker, "/STACK:204800000,204800000") 
#include <bits/stdc++.h>
using namespace std;
#define int long long
const int mod = 1e9 + 7;
const int N = 5e5 + 10;


int k, b, L, R;
int d;

int dp[200][200];
int a[200];
int len;

int qpow(int a, int b) {
    if (b == 0 && a == 0)return 0;
    int ans = 1;
    while (b) {
        if (b & 1) {
            ans *= a;
            ans %= mod;
        }
        a *= a;
        a %= mod;
        b >>= 1;
    }
    return ans;
}

int dfs(int pos, bool pre0, bool ismax, int sum) {
    if (pos == 0)return qpow(sum, k);
    if (!ismax && dp[pos][sum] != -1 && !pre0)return dp[pos][sum];
    int temp = 0;
    int maxn = ismax ? a[pos] - 1 : b - 1 ;
    if (ismax) {
        temp += dfs(pos - 1, 0, ismax, sum + (a[pos] == d)) % mod;
        temp %= mod;
        if (a[pos] == 0)return temp;
    }
    if (pre0) {
        temp += dfs(pos - 1, 1, 0, sum) % mod;
        temp %= mod;
    }
    else {
        temp += dfs(pos - 1, 0, 0, sum + (d == 0)) % mod;
        temp %= mod;
    }

    if (d != 0 && maxn >= d) {
        temp += dfs(pos - 1, 0, 0, sum + 1) % mod;
        temp %= mod;
        temp += (maxn - 1) % mod * dfs(pos - 1, 0, 0, sum) % mod;
        temp %= mod;
    }
    else {
        temp += (maxn) % mod * dfs(pos - 1, 0, 0, sum) % mod;
        temp %= mod;
    }
    if (!ismax && !pre0) {
        dp[pos][sum] = temp;
    }
    return temp;
}

void setNum(int num) {
    len = 0;
    while (num) {
        a[++len] = num % b;
        num /= b;
    }
}

int getAns(int num) {
    memset(a, 0, sizeof(a));
    len = 0;
    setNum(num);
    memset(dp, -1, sizeof(dp));
    return dfs(len, 1, 1, 0);
}

void solve() {
    cin >> k >> b >> d >> L >> R;
    cout << (getAns(R) - getAns(L - 1) + mod) % mod << '\n';
}


signed main() {
    int t;
    ios::sync_with_stdio(0);
    cin.tie(0);
    cout.tie(0);
    cin >> t;
    while (t --) {
        solve();
    }
}

总结

从这题可以看出，在解决计数类数位数位dp时，我们通常使用循环枚举这一位每种可能的结果，然后累加处理，但是其实在这么多种可能的数字中，返回值也就只有4种，我们可以通过将这4种结果进行分类讨论后进行累加，就可以大幅提高数位dp的效率。