【bzoj1257】[CQOI2007]余数之和sum

1257: [CQOI2007]余数之和sum

Time Limit: 5 Sec   Memory Limit: 162 MB
Submit: 4573   Solved: 2129
[ Submit][ Status][ Discuss]

Description

给出正整数n和k，计算j(n, k)=k mod 1 + k mod 2 + k mod 3 + … + k mod n的值，其中k mod i表示k除以i的余数。例如j(5, 3)=3 mod 1 + 3 mod 2 + 3 mod 3 + 3 mod 4 + 3 mod 5=0+1+0+3+3=7

Input

输入仅一行，包含两个整数n, k。

Output

输出仅一行，即j(n, k)。

Sample Input

5 3

Sample Output

7

HINT

50%的数据满足：1<=n, k<=1000 100%的数据满足：1<=n ,k<=10^9

Source

[ Submit][ Status][ Discuss]



研究了好久没有想出来。。

最后查了题解才有点启发

这一题首先要考虑到

对于k mod i,如果当i大于k时，对答案的贡献一定为k，这一部分可以直接先算

接下来考虑i <= k的时候

考虑到k mod i = k - trunc(k / i) * i

会发现有一些连续的i的trunc(k / i)的值会相等

这时候我们可以通过等差序列把这些trunc(k / i)相同的值求出来

于是我们就可以通过不断找这些trunc(k / i)相同的组，用二分查找这些组的右端点来完成i <= k情况的计算

那么复杂度呢？

我们会发现，trunc(k / i)的值的个数不会超过sqrt(k)个

为什么呢？

以sqrt(k)为分界点，trunc(k / i) (i <= sqrt(k))的取值肯定不会超过sqrt(k)个

而trunc(k / i) (i > sqrt(k))的取值肯定只有1一个

那么我们就可知，这些trunc(k / i)相同的组的个数不会超过sqrt(k)个

算上二分，复杂度O(sqrt(k) log k)

嗯。。感觉这种思想和分块有点像，姑且把它算作数学+分块吧

代码：

#include<cstdio>
#include<vector>
#include<queue>
#include<ctime>
#include<algorithm>
#include<cstdlib>
#include<stack>
#include<cstring>
#include<cmath>
using namespace std;

typedef long long LL;

const int INF = 2147483647;

LL pos,n,k,ans,maxx;

inline LL getint()
{
	LL ret = 0,f = 1;
	char c = getchar();
	while (c < '0' || c > '9')
	{
		if (c == '-') f = -1;
		c = getchar();
	}
	while (c >= '0' && c <= '9')
		ret = ret * 10 + c - '0',c = getchar();
	return ret * f;
}

inline LL binary()
{
	LL l = pos,r = maxx;
	while (r - l > 1)
	{
		LL mid = l + r >> 1;
		if (k / mid != k / pos)
			r = mid;
		else
			l = mid;
	}
	if (k / r == k / pos) return r;
	else return l;
}

int main()
{
	n = getint(); k = getint();
	maxx = min(n,k);
	pos = 1;
	while (pos <= maxx)
	{
		LL _pos = binary();
		LL tot = _pos - pos + 1;
		LL x = (_pos + pos) * tot / 2;
		ans += k * tot - k / pos * x;
		pos = _pos + 1;
	}
	if (n > k) ans += (n - k) * k;
	printf("%lld",ans);
	return 0;
}