bzoj 1257: [CQOI2007]余数之和

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。
1<=n ,k<=10^9

Output

输出仅一行，即j(n, k)。
Sample Input
5 3
Sample Output
7
//written by zzy

题目大意：

如题

题解：

k%i=k-i×|k/i|;
ans=n×k-sum(i×|k/i|)(i<=k)
考虑如何求i×k/i
推导：k/i在[i,k/|k/i|]区间内相等,为k/i
所以sum(i×|k/i|)在[i,k/|k/i|]区间内为k/i×sum(i);
因为i|k的数不超过2sqrt(k)
所以时间复杂度为O(sqrt(k))
只需不断更新区间即可。

#include<bits/stdc++.h>
using namespace std;
long long x,n,k,i,ans;
int main()
{
	scanf("%d%d",&n,&k);
	ans=n*k; 
	for (i=1,x;i<=n;i=x+1)
	{
		if (i>k) break;
		x=min(k/(k/i),n); 
		ans-=(k/i)*(i+x)*(x-i+1)/2;
	}
	printf("%lld",ans);
}