题目大意:
给出正整数n和k,计算j(n, k)=k mod 1 + k mod 2 + k mod 3 + … + k mod n 的值,1<=n,k<=10^9
【题解】
n>k的部分,总答案为 k*(n-k+1)n<=k的部分,总答案为 sigma(k mod i) = sigma(k-[k/i]*i) = sigma(k)-sigma([k/i]*i)
而i取某些连续值时,[k/i]连续,在这段区间中,sigma([k/i]*i) = [k/i]*sigma(i) 即:常数 * 等差数列之和
可以巧妙地求出这段连续区间,直接跳过
【代码】
#include<stdio.h>
#include<stdlib.h>
typedef long long LL;
LL min(LL a,LL b)
{
if(a<b) return a;
return b;
}
int main()
{
LL n,k,i,last,ans=0;
scanf("%lld%lld",&n,&k);
ans=n*k;
for(i=1;i<=k&&i<=n;i=last+1)
{
last=min(k/(k/i),n);
ans-=(k/i)*(last+i)*(last-i+1)/2;
}
printf("%lld",ans);
return 0;
}