算法竞赛进阶指南, 141页,数学推导
本题要点:
1、 k mod i = k - [k / i] * i, 答案就是 n * k - sum{[k / i] * i}(1 <= i <= n)
记 g(x) = [k / [k / x]], 可以证明 , [k / g(x)] == [k / x]
从而,x 在 区间[i, g(i)] 的范围内,[k / x] = [k / i],
也就是说, i <= x <= g(i)时候, [k / x] * x 形成了一个等差数列,
首项 [k / i] * i, 尾项 [k / i] * g(i), 公差为 [k / i]。 然后就是等差数列求和。
2、 区间 [1, k] 被分为理若干段, 每一段 [k / i] 的值是相等的。
求和 sum{[k / i] * i}(1 <= i <= n), 就是每一段的累加。
#include <cstdio>
#include <cstring>
#include <iostream>
using namespace std;
long long n, k;
int main()
{
scanf("%lld%lld", &n, &k);
long long ans = n * k;
if(n > k)
n = k;
long long i = 1, gx;
while(i <= n)
{
gx = k / (k / i);
if(gx > n)
{
gx = n;
}
ans -= ((gx - i + 1) * (gx + i) / 2) * (k / i);
i = gx + 1;
}
printf("%lld\n", ans);
return 0;
}
/*
10 5
*/
/*
29
*/