BZOJ1257 [CQOI2007]余數之和
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。
Output
輸出僅一行,即j(n, k)。
Sample Input
5 3Sample Output
7HINT
50%的數據滿足:1<=n, k<=1000 100%的數據滿足:1<=n ,k<=10^9
題解
首先,若$k<n$,那麽把答案加上$(n-k)*k$,然後n=k;
由於$k\mod i = k - \left\lfloor\frac k i\right\rfloor * i$,而有些$\left\lfloor\frac k i\right\rfloor$是相同的,可以一起計算,
即,若$\left\lfloor\frac k a\right\rfloor = \left\lfloor\frac k b\right\rfloor (a \leq b)$
那麽$[a,b]$對答案的貢獻是$k*(b-a) + \left\lfloor\frac k a\right\rfloor * \frac{(a+b)(b-a+1)}2$。
又因為對於某個數s,若存在$i$使得$\left\lfloor\frac k i\right\rfloor = s$,那麽最大的滿足條件的$i$為$\left\lfloor\frac k s\right\rfloor$,所以上述$[a,b]$的右端點$b$是可以確定的。
又因為$\left\lfloor\frac k i\right\rfloor$只有$O(\sqrt{k})$種($i \leq \sqrt{k}$時,只有$O(\sqrt{k})$個$i$;$i > \sqrt{k}$時,只有$O(\sqrt{k})$個$\left\lfloor\frac k i\right\rfloor$),所以復雜度為$O(\sqrt{k})$。
附代碼:
#include <cstdio> typedef long long LL; inline LL get(LL t) { return t * (t + 1) / 2; } int main() { LL n, k; scanf("%lld%lld", &n, &k); LL ans = k * n; if (n > k) n = k; for (LL i = 1, end; i <= n; i = end + 1) { end = k / (k / i); if (end > n) end = n; ans -= k / i * (get(end) - get(i - 1)); } printf("%lld\n", ans); return 0; }
BZOJ1257 [CQOI2007]余數之和