【ybtoj高效进阶6-2-5】余数之和/【luogu】P2261 [CQOI2007] 余数求和【 数论】【整除分块】

【ybtoj高效进阶6-2-5】余数之和 /【luogu P2261】 [CQOI2007] 余数求和

题目大意：

给定 $n$ , $k$，求$\sum _{i=1}^{n}k\mathrm{m}\mathrm{o}\mathrm{d}i$

思路：

先来分析一下整除分块这种东西
根据取模运算的性质，$x\mathrm{m}\mathrm{o}\mathrm{d}y=x-y\ast \lfloor \frac{x}{y}\rfloor$
则原式=$\sum _{i=1}^{n}k-i\ast \lfloor \frac{i}{k}\rfloor =n\ast k-\sum _{i=1}^{n}i\ast \lfloor \frac{i}{k}\rfloor$
右半部分可以使用整除分块处理

整除分块：

第一个结论：对于 $a,b,c\in Z$ $\lfloor \frac{a}{bc}\rfloor =\lfloor \frac{\lfloor \frac{a}{b}\rfloor }{c}\rfloor$
证明：
$\frac{a}{b}=\lfloor \frac{a}{b}\rfloor +x,x\in [0,1)$
$\because \lfloor \frac{a}{bc}\rfloor =\lfloor \frac{a}{b}\ast \frac{1}{c}\rfloor =\lfloor \frac{1}{c}\ast (\lfloor \frac{a}{b}\rfloor +x)\rfloor =\lfloor \frac{\lfloor \frac{a}{b}\rfloor }{c}+\frac{x}{c}\rfloor$

显然$\frac{x}{c}<1$， 对答案没有影响，直接舍去即可
然后就可以开始愉快的分块了

对于$\lfloor \frac{n}{i}\rfloor =\lfloor \frac{n}{j}\rfloor ,0<i\le j\le n$
这时候我们要找到最大的 $j$ 值
令 $k=\lfloor \frac{n}{i}\rfloor$，则有$k\le \frac{n}{i}$
$\therefore \lfloor \frac{n}{k}\rfloor \ge \lfloor \frac{n}{\frac{n}{i}}\rfloor =\lfloor i\rfloor =i$
$\therefore j=max,i_{max}=i=\lfloor \frac{n}{k}\rfloor =\lfloor \frac{n}{\lfloor \frac{n}{i}\rfloor }\rfloor$
$QED$
那么对于$\lfloor \frac{n}{i}\rfloor$相同的多个 $i$ 我们就可以打包计算了，$\lfloor \frac{n}{i}\rfloor$最多只有$\sqrt{n}$ 种情况（显然我不会证明）
那么整个算法的时间复杂度就是$O(\sqrt{n})$了

代码：

#include<iostream>
#include<cstdio>
#include<algorithm>
#include<queue>
#include<cstring>
#include<vector>
#include<cmath>
#include<ctime>
#define r register
#define rep(i,x,y) for(r ll i=x;i<=y;++i)
#define per(i,x,y) for(r ll i=x;i>=y;--i)
using namespace std;
typedef long long ll;
ll n,k; 
inline ll in()
{
	ll res=0,f=1;
	char ch;
	while((ch=getchar())<'0'||ch>'9')
	 if(ch=='-') f=-1;
	res=res*10+ch-48;
	while((ch=getchar())>='0'&&ch<='9')
	 res=res*10+ch-48;
	return res*f;
}
int main()
{
	n=in(),k=in();
	ll ans=n*k;
	for(r ll i=1,j;i<=n;i=j+1)
	{
		j=((k/i)==0)?n:min(n,k/(k/i)); //结论
		ans-=(k/i)*(j-i+1)*(i+j)>>1;//k/i 是值，(j-i+1)*（i+j）/2 等差数列的求和公式，好好理解
	}
	cout<<ans;
	return 0;
}