一、题目描述
二、算法分析说明与代码编写指导
如果直接暴力计算,由于n和k最大可达10^9,所以会超时。
但是随机取数k并将它模1到n的结果列出时,发现存在规律:常常出现取模结果相等的连续区间。
所以我们可以用除法分块的处理技巧进行优化:
当然,在本题中运用除法分块算法时还需要注意:
(所有分数线均代表整除,即C / C++中的 / 运算符)
所以直接套上面的除法分块算法求得k / i,i = 1,2,……,n的各取值的区间以后,计算时,每个区间的余数之和是:
注意求和的上限n,循环条件中要补充L<=n,否则部分数据在跑的过程会出现死循环,从而TLE。
三、AC代码
#include<cstdio>
#include<algorithm>
#pragma warning(disable:4996)
unsigned long long n, k, s, q;
int main() {
scanf("%llu%llu", &n, &k);
for (unsigned long long L = 1, R; L <= n; L = R + 1) {
q = k / L; if (q == 0)break;
R = std::min(k / q, n); s += q * (R - L + 1) * (L + R) / 2;
}
printf("%llu\n", n * k - s);
return 0;
}