【BZOJ 1257】[CQOI2007]余数之和sum

最新推荐文章于 2021-05-14 16:56:45 发布

原创最新推荐文章于 2021-05-14 16:56:45 发布 · 381 阅读

0 ·

CC 4.0 BY-SA版权

===数论=== 专栏收录该内容

18 篇文章

订阅专栏

本文介绍了一种计算从k%1到k%n所有余数之和的方法，通过数学推导简化了原始问题，并提供了一个高效的算法实现。适用于1≤n,k≤10^9的大规模输入。

题意

给出正整数 $n$ 和 $k$ ，计算 $j(n, k)=k \mod 1 + k \mod 2 + k \mod 3 + … + k \mod n$ 的值，其中 $k mod i$ 表示 $k$ 除以 $i$ 的余数。
$1\le n ,k\le 10^9$

思路

a n s = \sum i = 1 n k % i = \sum i = 1 n k - ⌊ k i ⌋ \times i = n k - \sum i = 1 n ⌊ k i ⌋ \times i

$ans=\sum_{i=1}^{n}k\%i=\sum_{i=1}^{n}k-\lfloor \frac{k}{i}\rfloor\times i = nk-\sum_{i=1}^{n}\lfloor \frac{k}{i}\rfloor\times i$
因为后面一项的前缀和可以维护，所以总的复杂度是

O(n−√) $O(\sqrt{n})$

代码

#include <iostream>
#include <cstdio>
#include <cmath>
using namespace std;
long long res, n, k, last;
int main(){
    scanf("%lld%lld", &n, &k);
    res += n*k;
    n = min(n, k);
    for(long long i = 1; i <= n; i = last+1){
        last = min(n, k/(k/i));
        res -= (k/i)*(i+last)*(last-i+1)/2;
    }
    printf("%lld", res); 
    return 0;
}