洛谷 P2261 [CQOI2007]余数求和

P2261 [CQOI2007]余数求和

题目链接-P2261 [CQOI2007]余数求和


解题思路
$除法分块$

• 我们很容易可以得出$amodb=a-b\ast \lfloor \frac{a}{b}\rfloor$
• 因为$G(n,k)={\sum }_{i=1}^{n}kmodi$，所以我们可以化简为$G(n,k)={\sum }_{i=1}^{n}k-i\ast \lfloor \frac{k}{i}\rfloor =n\ast k-{\sum }_{i=1}^{n}i\ast \lfloor \frac{k}{i}\rfloor$
• 我们通过打表找规律发现$\lfloor \frac{k}{i}\rfloor$在一定的区域内相等，显然$\lfloor \frac{k}{i}\rfloor$可以用除法分块来做
• 我们可以通过枚举$l$，然后再根据$l$和$k$计算出右边界$r$，很明显下一个块的左边界就为$r+1$，直到计算至$n$
• 然后分两种情况考虑：
      1. $\lfloor \frac{k}{l}\rfloor =0$时，$r=n$，（若$\lfloor \frac{k}{l}\rfloor =0$就意味着一直到结尾的区间内值都是零，所以令$r$直接等于$n$即可）
      2. $\lfloor \frac{k}{l}\rfloor \ne 0$时，$r=\lfloor \frac{k}{\lfloor \frac{k}{l}\rfloor }\rfloor$
• 接下来只需计算每个区间内$i\ast \lfloor \frac{k}{i}\rfloor$和即可，因为相同区间内$\lfloor \frac{k}{i}\rfloor$的值都为$k/l$，直接等差数列求${\sum }_l^{r}i$即可，即$(r+l)\ast (r-l+1)/2$，所以对于每个区间$ans$减去$(k/l)\ast (r+l)\ast (r-l+1)/2$即可
• 具体操作见代码

附上代码

#pragma GCC optimize("-Ofast","-funroll-all-loops")
#include<bits/stdc++.h>
#define int long long
#define lowbit(x) (x &(-x))
#define endl '\n'
using namespace std;
const int INF=0x3f3f3f3f;
const int dir[4][2]={-1,0,1,0,0,-1,0,1};
const double PI=acos(-1.0);
const double e=exp(1.0);
const double eps=1e-10;
const int M=1e9+7;
const int N=2e5+10;
typedef long long ll;
typedef pair<int,int> PII;
typedef unsigned long long ull;
inline void read(int &x){
    char t=getchar();
    while(!isdigit(t)) t=getchar();
    for(x=t^48,t=getchar();isdigit(t);t=getchar()) x=x*10+(t^48);
}
signed main(){
	ios::sync_with_stdio(false);
	cin.tie(0);cout.tie(0);

	int n,k;
	cin>>n>>k;
	int ans=n*k;
	for(int l=1,r;l<=n;l=r+1){
		if(k/l) r=min(n,k/(k/l));
		else r=n;
		ans-=(k/l)*(r-l+1)*(r+l)/2; 
	}
	cout<<ans<<endl;
	return 0;
}