【数论】【分块】【乱搞】codeforces 616E Sum of Remainders

E. Sum of Remainders

time limit per test

2 seconds

memory limit per test

256 megabytes

input

standard input

output

standard output

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.

Input

The only line contains two integers n, m (1 ≤ n, m ≤ 1013) — the parameters of the sum.

Output

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;
}