ATC DP S - Digit Sum（数位dp）

Godhui_

于 2022-03-01 23:46:43 发布

阅读量200

点赞数

分类专栏：动态规划文章标签：动态规划算法

本文链接：https://blog.csdn.net/Godhui_/article/details/123219340

版权

动态规划专栏收录该内容

15 篇文章 0 订阅

订阅专栏

ATC DP S - Digit Sum（数位dp）

S - Digit Sum

数位dp板子题，注释很详细了。

#include <iostream>
#include <cstdio>
#include <cstring>
#include <vector>
#include <set>
#include <unordered_map>
#include <cstring>

using namespace std;
const int N = 1e4 + 100;
const int mod = 1e9 + 7;
typedef long long ll;

string s;
int num[N];
ll f[N][105], D;
//f[i][j] 前i个数位,统计量为j的方案数

// 从前到后
// cur从高位到低位的第几位
// di_sum 当前数位和
// limit这位有没有限制
ll dfs(int cur, int di_sum, bool limit) {
    // 到第0位有数字和返回1
    if (cur == 0) {
        if (di_sum)return 0;
        else {
            return 1;
        }
    }
    // 如果当前位置搜过并且该数位不受限制直接返回
    if (f[cur][di_sum] != -1 && !limit)return f[cur][di_sum];

    ll res = 0;
    // 确定当前数位的上界限
    int up = limit ? num[cur] : 9;
    for (int i = 0; i <= up; i++) {
        int tem_sum = (di_sum - i + D) % D;
        int nx = cur - 1;
        bool nx_lim = limit && (i == up);
        res += dfs(nx, tem_sum, nx_lim);
        res %= mod;
    }
    // 如果当前位置是受限的 则不能完整表示后面xxxx9999
    if (!limit) f[cur][di_sum] = res;
    return res;
}

int main() {
    cin >> s;
    int len = s.length();
    for (int i = 1; i <= len; i++) {
        num[len - i + 1] = s[i - 1] - '0';
    }
    cin >> D;
    memset(f, -1, sizeof(f));
    ll res = dfs(len, 0, true);
    // 把0删了
    cout << (res - 1 + mod) % mod << endl;
    return 0;
}