题意:
给出n,k。求
n和k都是10^18,sqrt(k)理论上过不了,但是还是过了...
题挺好。
这题的分块和一般的分块不太一样,直接看图吧。
设k = d * x + r。发现每个块内都是一个等差数列,公差就是d。
于是我们枚举d,按d分块。
对于一个块[L, R](指的是i的区间),显然R = k / d,L用上一块的末尾加1,就是L = k / (d + 1) + 1。
然后就完了...
看代码。
(已知了d和x之后就不要用r = k % x了...用r = k - d * x更快。
注意当i大于k的时候要给r特判。
(下面这个少了个特判,但是可过)
#include <cstdio>
typedef unsigned long long ULL;
ULL n, k;
int main() {
scanf("%llu%llu", &n, &k);
ULL ans = 0;
for(ULL i = 1, l, r; i <= n; i = r + 1) {
ULL d = k / i;
l = k / (d + 1) + 1;
r = d ? k / d : n;
ans += (k - d * l + k - d * r) * (r - l + 1) / 2;
}
printf("%llu\n", ans);
}
999999999 1000000000000000000
这组数据我的代码跑了44s啊...看来数据水
2016.2.22 13:23补:
spoj数据太水啦!
做了BZOJ1257才知道,还要防止r大于n...
正确姿势:
#include <cstdio>
typedef unsigned long long ULL;
ULL n, k;
int main() {
scanf("%llu%llu", &n, &k);
ULL ans = 0;
for(ULL i = 1, l, r; i <= n; i = r + 1) {
ULL d = k / i;
l = k / (d + 1) + 1;
r = d ? k / d : n;
r = r > n ? n : r;
ans += (k - d * l + k - d * r) * (r - l + 1) / 2;
}
printf("%llu\n", ans);
}