题目描述
给出正整数 和 ,计算 的值。 。
算法分析
将取模运算变换形式: ,则 。
注意到随着 的递增, 的值递减,而且有一段部分相同,可以证明不同的 值有 种,考虑将这一部分一次性单独处理。
可以 计算出以 为起点的相同部分的终点 ,注意当分母 时会出错,这时应当特判终点为 ,还有后面乘 的部分直接使用等差数列公式求解即可,时间复杂度 。
实现代码
#include <cstdio>
#include <algorithm>
typedef long long int ll;
int main() {
ll n,k;
scanf("%lld%lld",&n,&k);
ll ans=n*k;
for(int i=1;i<=n;++i) {
ll end=k/i==0?n:std::min(n,k/(k/i));
ans-=(k/i)*(i+end)*(end-i+1)/2;
i=end;
}
printf("%lld\n",ans);
return 0;
}