题目入口
题目描述
给出正整数 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 !