Calculate the value of the sum: n mod 1 + n mod 2 + n mod 3 + ... + n mod m. As the result can be very large, you should print the value modulo 109 + 7 (the remainder when divided by 109 + 7).
The modulo operator a mod b stands for the remainder after dividing a by b. For example 10 mod 3 = 1.
The only line contains two integers n, m (1 ≤ n, m ≤ 1013) — the parameters of the sum.
Print integer s — the value of the required sum modulo 109 + 7.
题目链接:题目
我们可以通过转换,得到n%k=n-floor(n/k)*k。然后这个题目就变成了 sum=n*m-sigma(i:1 to m)(floor(n/i)*i);
佷明显,如果我们枚举i的话会TLE 枚举n/i的话也会TLE。
但是我们从另一个角度考虑,对于i小于sqrt(n)来说,i只有sqrt(n)种可能;
对于i>sqrt(n)来说,floor(n/i)也只有sqrt(n)种可能。。。
事实上这个题目也是这么搞得。
前面一种情况不多加赘述,一个for解决。
后面一种情况的话,我们枚举floor(n/i)那么我们就要知道i的最小取值和最大取值,如果知道的话,那么直接就等差数列相加再乘以floor(n/i)就好了。
此时如果对于floor(n/i)是递增枚举的话,那么最大取值就是上一个的最小取值减1。那么对于确定的floor(n/i),最小取值怎么计算呢。下面给予证明:
假设floor(n/i)=b,那么就有n/i=b余(i-k),要使i最小,所以就要k最小,且k大于0.然后转化过去,得到n+k=i*(b+1),再转化得到i=(n+k)/b,把k变成1+t。
变成i=(n+1+t)/b。这个时候,我们可以得到i最小其实就是ceil((n+1)/b)。防止误差可以写成n/b+1;
到这里,全部说完了,全程注意相乘爆long long。这道题目就这么完了。
代码如下:
#include<iostream>
#include<algorithm>
using namespace std;
const long long MOD=1000000007;
int main()
{
long long m,n;
cin>>n>>m;
long long mm=m;
m=min(m,n);
long long ans=0;
long long mins=n/m;
long long maxs=sqrt(n);
long long last=m; //记录上一次的min-1
<span style="white-space:pre"> </span>//先枚举比较麻烦的部分
while(mins<=maxs)
{
long long minss=(n)/(mins+1)+1;
long long maxss;
maxss=last;
minss=min(minss,last);
long long tmp=maxss-minss+1,tmp1=minss+maxss;
if(tmp%2==0)
tmp/=2;
else
tmp1/=2;
<span style="white-space:pre"> </span>//注意爆long long
tmp%=MOD;
tmp1%=MOD;
ans+=mins*tmp%MOD*tmp1%MOD;
ans%=MOD;
mins++;
last=minss-1;
}
long long x=last;
for(long long int i=1;i<=x;i++)
{
ans+=n/i*i;
ans%=MOD;
}
cout<<(mm%MOD*(n%MOD)%MOD-ans+MOD)%MOD<<endl;
return 0;
}