P2261 [CQOI2007]余数求和——解题报告

一、题目链接

P2261 [CQOI2007]余数求和

二、题目大意

给定正整数$n$，$k$，要求计算${\sum }_{i=1}^{n}kmodi$。
数据范围：$1\le n,k\le 10^9$

三、题目分析

如果直接求解这个问题，它的时间复杂度就是$O(n)$，不满足题目的要求，所以我们要首先分析这个运算的本质。

根据取余运算的特性，我们知道$kmodi=k-\lfloor k/i\rfloor \times i$，所以，我们要求的东西就变成了$n\times k-{\sum }_{i=1}^n\lfloor k/i\rfloor \times i$。

我们很容易发现，在当$i$为某些值的时候，$\lfloor k/i\rfloor$的值是相同的，我们考虑如何将这些相同的数合在一起计算，降低时间复杂度。

• 我们设$temp=\lfloor k/\lfloor k/x\rfloor \rfloor$，$temp\ge x$，所以$\lfloor k/temp\rfloor \le \lfloor k/x\rfloor$
• 又因为$temp\le k/\lfloor k/x\rfloor =ano$，所以$\lfloor k/temp\rfloor \ge \lfloor k/ano\rfloor$
• 显然，$k/ano=\lfloor k/x\rfloor$。即$\lfloor k/temp\rfloor \ge \lfloor k/x\rfloor$

所以我们知道$\lfloor k/temp\rfloor =\lfloor k/x\rfloor$。即在$i\in [x,\lfloor k/\lfloor k/x\rfloor \rfloor ]$时$\lfloor k/i\rfloor$相等。

• 于是我们发现，在同一个区间，即$\lfloor k/x\rfloor$不变的时候。当$i$变化时，$\lfloor k/x\rfloor \ast i$就形成了一个等差数列，我们可以直接用等差数列求和公式来计算这一段的答案，时间复杂度为$O(1)$。
• 接下来考虑有多少个区间需要计算
  • 当$i<\sqrt{k}$时，因为除数只有i种的限制，显然结果最多只有$\sqrt{k}$种。
  • 当$i>\sqrt{k}$时，$\lfloor k/i\rfloor <\sqrt{k}$显然成立，所以也只有$\sqrt{k}$种。

综上所述，区间个数的数量级在$\sqrt{k}$，所以时间复杂度为$O(\sqrt{k})$。

四、正解程序

#include <iostream>
#include <cstdio>
#include <cstring>
#include <cstdlib>
#include <algorithm>
#include <cmath>

using namespace std;
typedef long long ll;

int main()
{
	ll n,k,ans,gx;
	scanf("%lld%lld",&n,&k);
	ans=n*k;
	if(k==0)
	{
		printf("%lld\n",ans);
		return 0;
	}
	for(ll x=1;x<=n;x=gx+1)
	{
		gx=k/x?min(k/(k/x),n):n;
		ans-=(k/x)*(x+gx)*(gx-x+1)/2;
	}
	printf("%lld\n",ans);
	
	return 0;
}