LOJ-1068 Investigation（数位dp）

最新推荐文章于 2023-04-02 00:07:35 发布

JXNU_SONG

最新推荐文章于 2023-04-02 00:07:35 发布

阅读量172

点赞数

分类专栏： DP

本文链接：https://blog.csdn.net/qq_45863710/article/details/118928240

版权

DP 专栏收录该内容

23 篇文章 2 订阅

订阅专栏

Investigation

原题链接

题目描述：对于每个样例给出 $l, r, k$ ，问在 $[l, r]$ 中，有多少数字可以被 $k$ 整除且其数位和也被 $k$ 整除。

考虑数位 $d p$ ，我们首先想像普通的数位 $d p$ 一样，可以维护一个当前数字 $mod\ k$ 和数位和 $mod\ k$ 两个状态，但是我们发现这个 $k$ 很大，但是我们又可以发现一个问题，给出的 $l, r$ 是在 $i n t$ 范围内的，这个范围内的最大数位和也才 $82$ ，且不会有数位和为 $0$ 的情况出现，那么当 $k > 82$ 的时候，任何一个范围内的所有数字的数位和 $mod\ k$ 都会等于其数位和本身，且不可能为 $0$ ，所以当 $k > 82$ 的时候，就直接输出 $0$ 即可，小于等于的话我们就可以用数位 $d p$ 解决。第一维表示当前放第几位，第二维表示是否有最高位限制，第三维表示当前数字和 $mod\ k$ ，第四维表示当前数位和 $mod\ k$ 。

#include <bits/stdc++.h>

using namespace std;

int l, r, k;
int dp[10][2][90][90];
vector<int> num;

int dfs(int pos, int limit, int smod, int dsmod) {
    if (pos < 0) return (smod == 0 && dsmod == 0);
    if (~dp[pos][limit][smod][dsmod]) return dp[pos][limit][smod][dsmod];
    int Max = limit ? num[pos] : 9, ans = 0;
    for (int i = 0; i <= Max; ++i) {
        ans += dfs(pos - 1, limit && i == num[pos], (10 * smod + i) % k, (dsmod + i) % k);
    }
    dp[pos][limit][smod][dsmod] = ans;
    return ans;
}

int cal(int x) {
    memset(dp, -1, sizeof dp);
    num.clear();
    while(x) {
        num.push_back(x % 10);
        x /= 10;
    }
    return dfs((int)num.size() - 1, 1, 0, 0);
}

int main() {
#ifndef ONLINE_JUDGE
    freopen("in.txt", "r", stdin);
    freopen("out.txt", "w", stdout);
#endif
    int T, cnt = 0;
    scanf("%d", &T);
    while(T--) {
        scanf("%d%d%d", &l, &r, &k);
        if (k > 82) {
            printf("Case %d: 0\n", ++cnt);
        }
        else {
            printf("Case %d: %d\n", ++cnt, cal(r) - cal(l - 1));
        }
    }
}