【BZOJ 1257】[CQOI2007]余数之和sum

题意

给出正整数$n$和$k$，计算$j(n, k)=k \mod 1 + k \mod 2 + k \mod 3 + … + k \mod n$的值，其中$k mod i$表示$k$除以$i$的余数。
$1\le n ,k\le 10^9$

思路

$$ans=\sum_{i=1}^{n}k\%i=\sum_{i=1}^{n}k-\lfloor \frac{k}{i}\rfloor\times i = nk-\sum_{i=1}^{n}\lfloor \frac{k}{i}\rfloor\times i$$
因为后面一项的前缀和可以维护，所以总的复杂度是 $O(\sqrt{n})$

代码

#include <iostream>
#include <cstdio>
#include <cmath>
using namespace std;
long long res, n, k, last;
int main(){
    scanf("%lld%lld", &n, &k);
    res += n*k;
    n = min(n, k);
    for(long long i = 1; i <= n; i = last+1){
        last = min(n, k/(k/i));
        res -= (k/i)*(i+last)*(last-i+1)/2;
    }
    printf("%lld", res); 
    return 0;
}