educational round #5 cf616E Sum of Remainders

询问 \(n\mod 1+n\mod 2+n\mod 3+\cdots +n\mod m\) 对 \(10^9+7\) 取模的值.

\(1\leq n,m\leq 10^{13}\)

考虑 \(\mathrm O(m)\) 地计算肯定是不可行的 .

是否可以和分解因数一样,只考虑 \(\mathrm O(\sqrt n)\) 之内的内容呢 ?

是可以的,观察可得对于 \(1\leq x\leq \sqrt n\) ,对于 \(\lfloor\frac{n}{y}\rfloor =x\) 的 \(y\) 的范围是 $\lfloor \frac{n}{x+1}\rfloor+1 \leq y\leq \lfloor \frac{n}{x}\rfloor $ .

那么,对于这些 \(y\) 的模数是 \(n-x\lfloor \frac{n}{x}\rfloor\) 到 \(n-x(\lfloor \frac{n}{x+1}\rfloor+1)\) ,依次递增 \(x\) . 这个直接等差序列求和 .

这是,可以发现,对于 \(m=x\) 和 \(m\in [\lfloor \frac{n}{x+1}\rfloor+1 ,\lfloor \frac{n}{x}\rfloor ]\) 是可以 \(\mathrm O(1)\) 地求和的 .

随着 \(x\) 的增加,可以求和的区间形如两边向中间衍生,知道相遇合并 .

所以,发现,只要枚举 \(x\in [1,\sqrt n]\) ,就可以得到所有 \([1,m]\) 的所有模数.

可能要特殊处理一下最后 \(x=\sqrt n\) 的情况 .

此题要非常注意取模 .

时间复杂度 : \(\mathrm O(\sqrt n)\)

空间复杂度 : \(\mathrm O(1)\)

code

#include<bits/stdc++.h>
using namespace std;
inline long long read(){
	char ch=getchar();
	while(ch<'0'||ch>'9')ch=getchar();
	long long res=0;
	while(ch>='0'&&ch<='9'){
		res=(res<<3)+(res<<1)+ch-'0';
		ch=getchar();
	}
	return res;
}
inline void print(int res){
	if(res==0){
		putchar('0');
		return;
	}
	int a[10],len=0;
	while(res>0){
		a[len++]=res%10;
		res/=10;
	}
	for(int i=len-1;i>=0;i--)
		putchar(a[i]+'0');
}
const int mod=1e9+7;
long long n,m;
long long ans=0;
int main(){
	int inv2=(mod+1)/2;
	n=read();m=read();
	long long s=0;
	while(s*s<=n)s++;
	if(s*s>n)s--;
	if(m<=s){
		for(int i=1;i<=m;i++){
			ans=(n%i+ans)%mod;
		}
		print(ans);
		putchar('\n');
		return 0;
	}
	int ans=0;
	for(long long i=1;(i+1)*(i+1)<=n;i++){
		ans=(ans+n%i)%mod;
		long long r=n/i,l=n/(i+1)+1;
		if(l>min(n-1,m))continue;
		r=min(r,min(n-1,m)); 
		long long R=n-r*i,L=n-l*i;
		long long tmp=1ll*(R+L)%mod*((r-l+1)%mod)%mod*inv2%mod; 
		ans=(ans+tmp)%mod;
	}
	if(1ll*s*s==n){
		if(s<=m){
			ans=(ans+n%s)%mod;	
		}
	}else{
		long long r=n/s;
		r=min(r,min(n-1,m));
		long long l=s+1; 
		if(l>r)l=r;
		long long R=n-r*s,L=n-l*s;
		long long tmp=1ll*(R+L)%mod*((r-l+1)%mod)%mod*inv2%mod;
		ans=(ans+tmp)%mod;
		if(l!=s)ans=(ans+n%s)%mod;
	}
	if(m>n)ans=(1ll*(m-n)%mod*(n%mod)%mod+ans)%mod;
	print(ans);
	putchar('\n');
	return 0;
}
/*inline? ll or int? size? min max?*/
上一篇:欧拉函数+反演——2019hdu多校6588


下一篇:500行代码,教你用python写个微信飞机大战