【bzoj 1257】余数之和sum（数学）

传送门biu~
对于原式我们可以考虑先计算$n\times k$，再把$mod$掉的部分减去。当$k\ mod\ i$时，减去的部分相当于$(k/i)\times i$。而对于一些连续的$i$，$(k/i)$可能是相同的。所以对这部分连续的$i$我们可以先求出它的区间大小$[l,r]$，然后用等差数列求和公式计算。因为$(k/i)$的值一共有$\sqrt n$种，时间复杂度$O(\sqrt n)$。

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