BZOJ 1257: [CQOI2007]余数之和【分块】

1257: [CQOI2007]余数之和

Time Limit: 5 Sec Memory Limit: 128 MB

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

题解

我们可以把k mod i看成k-(k/i)*i。
所以就发现k/i有相同的几段，那么就是分块，然后用等差数列求相同的一块。

代码如下

#include<cstdio>
#define LL long long
using namespace std;
int n,k;
LL ans;
int main(){
    scanf("%d%d",&n,&k);
    if(n>k) ans+=(LL)(n-k)*k,n=k;
    for(int i=1;i<=n;i++){
        int l,r,len,t=k/i;l=i;r=k/t;
        if(r>n) r=n;
        len=r-l+1;ans+=((LL)k-t*l+k-t*r)*len/2;i=r;
    }
    printf("%lld\n",ans);
    return 0;
}