Description
给出正整数n和k,计算j(n, k)=k mod 1 + k mod 2 + k mod 3 + … + k mod n的值,其中k mod i表示k除以i的余数。例如j(5, 3)=3 mod 1 + 3 mod 2 + 3 mod 3 + 3 mod 4 + 3 mod 5=0+1+0+3+3=7
Solution
比较简单
i>k的部分,取模值相同,答案为 ((n-k)*k)
i<k的部分,分块做,(k/i) 相同的一组,会发现是一个以 (k/i) 为公差的等差数列,直接算即可
#include <algorithm>
#include <iostream>
#include <cstdlib>
#include <cstring>
#include <cstdio>
#include <cmath>
#define RG register
#define il inline
#define iter iterator
#define Max(a,b) ((a)>(b)?(a):(b))
#define Min(a,b) ((a)<(b)?(a):(b))
using namespace std;
typedef long long ll;
void work()
{
ll n,k,r;
cin>>n>>k;
ll t1,t2,ans=0,lim=min(n,k),x;
for(int i=1;i<=lim;i=r+1){
r=min(k/(k/i),n);
t1=k%i;t2=k%r;
x=(t1-t2)/(k/i);
ans+=(x+1)*(t1+t2)/2;
}
if(n>k)ans+=(n-k)*k;
cout<<ans<<endl;
}
int main()
{
work();
return 0;
}