首先,我们先来看以下一道题。
求
∑
i
=
1
n
⌊
n
i
⌋
\sum\limits_{i=1}^n\lfloor\dfrac ni\rfloor
i=1∑n⌊in⌋
如果用朴素算法,则需要 O ( n ) O(n) O(n)的时间复杂度。那有没有更快的方法呢?
我们可以分情况考虑:
- 当 1 ≤ i ≤ n 1\leq i\leq \sqrt n 1≤i≤n时,因为 i i i只能取 n \sqrt n n个值, ⌊ n i ⌋ \lfloor \dfrac ni\rfloor ⌊in⌋最多只能取 n \sqrt n n个值
- 当 n ≤ i ≤ n \sqrt n\leq i\leq n n≤i≤n时, 1 ≤ n i ≤ n 1\leq \dfrac ni\leq \sqrt n 1≤in≤n,所以此时 ⌊ n i ⌋ \lfloor \dfrac ni\rfloor ⌊in⌋最多取 n \sqrt n n个值
也就是说, ⌊ n i ⌋ \lfloor \dfrac ni\rfloor ⌊in⌋总共能取 2 n 2\sqrt n 2n个值。又因为 ⌊ n i ⌋ \lfloor \dfrac ni\rfloor ⌊in⌋随着 i i i增大而减小,所以相同的 ⌊ n i ⌋ \lfloor \dfrac ni\rfloor ⌊in⌋值在同一个区间。
举个例子,若 n = 10 n=10 n=10,则
i i i | 1 1 1 | 2 2 2 | 3 3 3 | 4 4 4 | 5 5 5 | 6 6 6 | 7 7 7 | 8 8 8 | 9 9 9 | 10 10 10 |
---|---|---|---|---|---|---|---|---|---|---|
⌊ n i ⌋ \lfloor \dfrac ni \rfloor ⌊in⌋ | 10 10 10 | 5 5 5 | 3 3 3 | 2 2 2 | 2 2 2 | 1 1 1 | 1 1 1 | 1 1 1 | 1 1 1 | 1 1 1 |
对于一个有相同 ⌊ n i ⌋ \lfloor \dfrac ni\rfloor ⌊in⌋值的区间,我们考虑在知道其左端点 l l l的情况下,如何快速地求出其右端点。显然,这个区间中所有 ⌊ n i ⌋ \lfloor \dfrac ni\rfloor ⌊in⌋的值都为 ⌊ n l ⌋ \lfloor \dfrac nl\rfloor ⌊ln⌋。令 d = ⌊ n l ⌋ d=\lfloor \dfrac nl\rfloor d=⌊ln⌋,那么有
d = ⌊ n i ⌋ = n − n % i i i × d = n − n % i ≤ n i ≤ n d d=\lfloor \dfrac ni\rfloor=\dfrac{n-n\%i}{i} \\ \qquad \\i\times d=n-n\% i\leq n \\ \qquad \\i\leq \dfrac nd d=⌊in⌋=in−n%ii×d=n−n%i≤ni≤dn
因为 r r r为右端点,所以 r r r是最大的, r = ⌊ n d ⌋ = ⌊ n ⌊ n l ⌋ ⌋ r=\lfloor\dfrac nd\rfloor=\lfloor\dfrac{n}{\lfloor \frac nl\rfloor}\rfloor r=⌊dn⌋=⌊⌊ln⌋n⌋
由上文, ⌊ n i ⌋ \lfloor \dfrac ni\rfloor ⌊in⌋最多只能取 2 n 2\sqrt n 2n个值,所以时间复杂度为 O ( n ) O(\sqrt n) O(n)。
code
#include<bits/stdc++.h>
using namespace std;
int n,ans=0;
int main()
{
scanf("%d",&n);
for(int l=1,r;l<=n;l=r+1){
r=n/(n/l);
ans+=(r-l+1)*(n/l);
}
printf("%d",ans);
return 0;
}
类似地,下面的题也可以用数论分块解决。
求
∑
i
=
1
min
(
n
,
m
)
⌊
n
i
⌋
−
⌊
m
i
⌋
\sum\limits_{i=1}^{\min(n,m)}\lfloor \dfrac ni\rfloor -\lfloor\dfrac mi\rfloor
i=1∑min(n,m)⌊in⌋−⌊im⌋
code
#include<bits/stdc++.h>
using namespace std;
int n,m,ans=0;
int main()
{
scanf("%d%d",&n,&m);
for(int l=1,r;l<=min(n,m);l=r+1){
r=min(n/(n/l),m/(m/l));
ans+=(r-l+1)*(n/l-m/l);
}
printf("%d",ans);
return 0;
}
设 f , g f,g f,g为数论函数, F F F为 f f f的前缀和。若 F , g F,g F,g已知,则如下式子也可以用数论分块来求。
∑ i = 1 n f ( i ) × g ( ⌊ n i ⌋ ) \sum\limits_{i=1}^nf(i)\times g(\lfloor\dfrac ni\rfloor) i=1∑nf(i)×g(⌊in⌋)
对于每一块,设左右端点分别为 l , r l,r l,r,则这一块的贡献为
( F ( r ) − F ( l − 1 ) ) × g ( ⌊ n l ⌋ ) (F(r)-F(l-1))\times g(\lfloor\dfrac nl\rfloor) (F(r)−F(l−1))×g(⌊ln⌋)
时间复杂度为 O ( n ) O(\sqrt n) O(n)。