BZOJ1257 CQOI2007 余数之和 【数分块】
BZOJ1257 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
这世界上居然还有这么简单的省选题
式子太简单不讲了
#include<bits/stdc++.h>
using namespace std;
#define LL long long
LL n,k,ans=0;
int main(){
scanf("%d%d",&n,&k);
ans+=n*k;
if(n>=k)ans-=k;//n==k
n=min(n,k-1);
for(LL i=1,j;i<=n;i=j+1){
j=min(k/(k/i),n);
ans=ans-(i+j)*(j-i+1)*(k/i)/2;
}
printf("%lld",ans);
return 0;
}