BZOJ 1257 数学 思维 分段

题目链接

题意:

给出正整数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.

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

思路:

这个题目,数据范围这么大,肯定有巧妙的方法。

=∑k mod i, 1<=i<=n
=∑k-[k/i]*i，1<=i<=n
=n*k-∑[k/i]*i，1<=i<=n

当i>k时[k/i]为0，所以只需要考虑i<=k的情况

根据性质知道[k/i]的取值个数不超过根号k个。

且相同的[k/i]的取值的作为一个连续区间，是一个等差数列.那么我们只需要确定这个连续的区间,然后计算和即可.

怎么样去计算区间呢？

用i从1到min(n,k)枚举，每次找到取值w，算出左右区间。
左区间：L=i。
右区间：当[k/i]=w时，
根据高斯消元的性质，w=[k/i]<=k/i，
∴i<=k/w，R=(int)k/w。那么我们就找到了这个等差区间,然后求和即可.
∵可能存在R>=n，所以R=min(R,n)。
对于每个区间，res+=w*∑q，L<=q<=R。
然后i=R+1。

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int mod=1e9+7;
const int maxn=1e5+10;
ll n,k;
int main(){

	scanf("%lld%lld",&n,&k);
	ll ans=n*k;
	n=min(n,k);
	ll r,cnt,w; 
	for(int i=1;i<=n;i=r+1)
	{
		 w=k/i;
		 r=k/w;
		 if(r>n)
		 r=n;
		 cnt=r-i+1;
		ans-=w*cnt*(i+r)/2;
	}
	printf("%lld\n",ans);
	return 0;
}