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
Input
输入仅一行,包含两个整数n, k。
Output
输出仅一行,即j(n, k)。
Sample Input
5 3
Sample Output
7
HINT
50%的数据满足:1<=n, k<=1000 100%的数据满足:1<=n ,k<=10^9
思路
这题呀其实很简单
首先我们看如果
n>k
那么 k%i==k
我们直接将n>k的部分加上(n-k)*k并将n变成k-1
然后,再1~min(k-1,n)的区间里我们可以得到一个规律,在
k/l==k/r
的这个范围内,模数是一个等差数列
知道这个性质之后,我们就可以二分来搞定对于每一个k/l的结尾,然后再搞一搞就好了
代码
#include <bits/stdc++.h>
#define ll long long
#define int long long
using namespace std;
inline int read(){
int ret=0,f=1;char c=getchar();
for(;!isdigit(c);c=getchar())if(c=='-')f=-1;
while(c>='0'&&c<='9')ret=ret*10+c-'0',c=getchar();
return ret*f;
}
int n,k;
unsigned long long ans=0ll;
#undef int
int main(){
#define int long long
n=read();k=read();
if(n>k)ans=(n-k)*k;
if(n>k-1)n=k-1;
ll now=1;
while(now<=n){
ll l=now,r=n+1;
while(l<r-1){
ll mid=(l+r)>>1;
if(k/mid==k/now)l=mid;
else r=mid;
}
ans+=(l-now+1)*(k%now+k%l)/2;
now=l+1;
}
cout<<ans;
return 0;
}