AtCoder Beginner Contest 336 E - Digit Sum Divisible

最新推荐文章于 2024-06-23 22:33:26 发布

吵闹的人群保持笑容多冷静

最新推荐文章于 2024-06-23 22:33:26 发布

阅读量1.2k

点赞数 20

分类专栏： AtCoder 文章标签：算法 c++ 笔记动态规划

本文链接：https://blog.csdn.net/m0_73500785/article/details/135654850

版权

AtCoder 专栏收录该内容

7 篇文章 1 订阅

订阅专栏

E - Digit Sum Divisible

题意

定义一个正整数 $x$ 为 $g oo d$ 当且仅当： $x$ 能被它的数位和整除

统计小于等于 $N$ 的 $g oo d$ 正整数数量

思路

这道题关键是要观察到在 $\leq 10^{14}$ 的限制下，数位和种类是有限的：最多只有 $\times \log_{10}10^{14} = 9 \times 14 = 126$ 种。那么我们可以枚举每一种数位和为 $m o d$ ，看看有多少正整数 $x$ 可以被它整除，也就是 $x$ % $m o d = 0$

很明显这些涉及到数位统计并且还要计数的问题，我们可以使用数位 $D P$ 来计算。关键是怎么定义 $D P$ 状态，还有高位如何利用低位的计算结果。

很显然，对于一个固定的数位和 $m o d$ ，我们 $D P$ 的状态限制有：
$p os （低 p os 位是全变化位）$ 、
$s u m （高位的已经确定的数位的和）$ 、
$r （高位的已经确定的数位单独构成一个数字时模除 m o d 的余数）$

例如： $p os = 2, s u m = 6, r = 3, m o d = 20$ 时，代表的是 $\rightarrow 123 '99'$ 这些数字。因为这些数字的低 $p os = 2$ 位是全变化位，也就是从 $0$ ~ $9$ 任意取的位；更高的位的数位和为 $1 + 2 + 3 = 6 = s u m$ ；高位已经确定的位单独构成数字 $123$ 时，模除 $m o d = 20$ 的余数是 $r = 3$ 。（当然，这里只是举一个例子，在上述限制条件下，可能还有别的数字符合条件但没有列举出来）

有了限制条件，现在我们考虑如何转移。
根据上述限制我们有定义： $d p [p os] [s u m] [r]$ 为限制条件下 $g oo d$ 的正整数数量。如果 $N$ 的长度为 $l e n$ ，那么我们当前的状态下： $p_{len}p_{len-1}p_{len-2} ...p_{pos+1}p_{pos}p_{pos-1}...p_2p_1$ ，有：

$sum = p_{len} + p_{len-1} + ... + p_{pos+1}$
$(10^{len-1} \cdot p_{len} + 10^{len-2} \cdot p_{len-1} + ... + 10^{pos} \cdot p_{pos+1})$ % $m o d$

那么如果我们枚举当前第 $p os$ 位为 $i$ 的话，新的状态：
$sum\prime= sum + i$
$r\prime = (10 \times r + i)$ % $m o d$

这里关键是观察到 $r\prime$ 的变化，通过前面更高位左移（整体 $×10 \times 10$ ）得到。

这样我们就完成了状态转移，记忆化搜索到最底层的时候，判断 $s u m = m o d$ 并且 $r = 0$ 时才返回 $1$

时间复杂度：共有 $\times 14 = 126$ 种数位和，范围 $N$ 长度为 $l e n$ ，每种数位和要搜索的范围可以通过 $D P$ 数组得出，大约为 $len \times D \times mod$
故最终复杂度 $O(D^3 \cdot len)$

#include<bits/stdc++.h>
#define fore(i,l,r)	for(int i=(int)(l);i<(int)(r);++i)
#define fi first
#define se second
#define endl '\n'
#define ull unsigned long long
#define ALL(v) v.begin(), v.end()
#define Debug(x, ed) std::cerr << #x << " = " << x << ed;

const int INF=0x3f3f3f3e;
const long long INFLL=1e18;

typedef long long ll;

ll dp[17][150][150]; //dp[pos][sum][r] 表示低pos位为全变化位，前面高位已经确定的数位和为sum，高位模mod为r的数量
int num[17];
int mod;

ll dfs(int pos, int sum, int r, bool limit){
    if(!pos) return !r && sum == mod; //最底层，这时候数位和一定要是mod，并且sum % mod 要为 0
    if(!limit && ~dp[pos][sum][r]) return dp[pos][sum][r];
    ll res = 0;
    int up = (limit ? num[pos] : 9);
    fore(i, 0, up + 1){
        res += dfs(pos - 1, sum + i, (r * 10 + i) % mod, limit && i == up);
    }
    if(!limit) dp[pos][sum][r] = res;
    return res;
}

ll solve(ll x){
    int len = 0;
    while(x){
        num[++len] = x % 10;
        x /= 10;
    }
    ll ans = 0;
    for(mod = 1; mod < 130; ++mod){ // 枚举整个数位和
        memset(dp, -1, sizeof(dp));
        ans += dfs(len, 0, 0, true);
    }
    return ans;
}

int main(){
    std::ios::sync_with_stdio(false);
    std::cin.tie(nullptr);
    std::cout.tie(nullptr);
    ll n;
    std::cin >> n;
    std::cout<< solve(n);
    return 0;
}