洛谷P2261(余数求和)

题目入口
题目描述
给出正整数 n 和 k 计算 G(n, k) = k mod 1 + k mod 2 + k mod 3 + ……+ k mod n 的值 其中 k mod i 表示 k 除以 i 的余数。

例如 G(10, 5)= 5 mod 1 + 5 mod 2 + 5 mod 3 + 5 mod 4 + 5 mod 5 + …… + 5 mod 10 = 0 + 1 + 2 + 1 + 0 + 5 + 5 + 5 + 5 + 5 = 29.

输入格式
两个整数 n ，k

输出格式
答案

输入 #1
10 5

输出 #1
29

说明/提示
30%: n , k ≤ 1000
60%: n , k ≤ 10 ^ 6
100%: n , k ≤ 10 ^ 9

思路：显然数据很多，简单的遍历是不可取的。题目本本质是需要求出 ans = ∑(n) (i=1) k%i；但是同理我们也知道，a%b = a - b * [a/b]，所以也可得ans = n * k - ∑(n) (i=1) i*[k/i]。
因为[k/i]大于零的有√k种，所以时间复杂度为O(√k).

这利用的就是分块的思维：
1 2 3 4 5 6 7 8 9 10 <- i
5 2 1 1 1 0 0 0 0 0 <- [k/i]
1 2 5 5 5 - - - - - <- r
eg:
i = 1: ans -= 5;
i = 2: ans -= 4;
i = 3: ans -= 12 ; break;

可见[k/i]可分成3块，分别ans减去三块的和就行了。
在循环遍历(块)的时候需要另设一个变量r，让每次i = r +1
if ([k/i] == 0 ) r = n;
else r = min(n, (k / [k/i]));
而每一块的和即为 [k/i] * (r - i + 1) * (r + i) / 2;
如此遍历√k次就能得到答案啦。

代码实现：

#include<bits/stdc++.h>
#define int long long
using namespace std;
int n, k, ans;
signed main()
{
    cin >> n >> k;
    ans = n * k;
    for(int i = 1, r; i <= n; i = r + 1){
        if(k / i) r = min(n, k / (k / i));
        else r = n;
        ans -= (k / i) * (r - i + 1) * (r + i) / 2;
    }
    cout << ans << endl;
    return 0;
}

得注意开 long long !