首先,我们先来看以下一道题。
求
∑i=1n⌊ni⌋\sum\limits_{i=1}^n\lfloor\dfrac ni\rfloori=1∑n⌊in⌋
如果用朴素算法,则需要O(n)O(n)O(n)的时间复杂度。那有没有更快的方法呢?
我们可以分情况考虑:
- 当1≤i≤n1\leq i\leq \sqrt n1≤i≤n时,因为iii只能取n\sqrt nn个值,⌊ni⌋\lfloor \dfrac ni\rfloor⌊in⌋最多只能取n\sqrt nn个值
- 当n≤i≤n\sqrt n\leq i\leq nn≤i≤n时,1≤ni≤n1\leq \dfrac ni\leq \sqrt n1≤in≤n,所以此时⌊ni⌋\lfloor \dfrac ni\rfloor⌊in⌋最多取n\sqrt nn个值
也就是说,⌊ni⌋\lfloor \dfrac ni\rfloor⌊in⌋总共能取2n2\sqrt n2n个值。又因为⌊ni⌋\lfloor \dfrac ni\rfloor⌊in⌋随着iii增大而减小,所以相同的⌊ni⌋\lfloor \dfrac ni\rfloor⌊in⌋值在同一个区间。
举个例子,若n=10n=10n=10,则
| iii | 111 | 222 | 333 | 444 | 555 | 666 | 777 | 888 | 999 | 101010 |
|---|---|---|---|---|---|---|---|---|---|---|
| ⌊ni⌋\lfloor \dfrac ni \rfloor⌊in⌋ | 101010 | 555 | 333 | 222 | 222 | 111 | 111 | 111 | 111 | 111 |
对于一个有相同⌊ni⌋\lfloor \dfrac ni\rfloor⌊in⌋值的区间,我们考虑在知道其左端点lll的情况下,如何快速地求出其右端点。显然,这个区间中所有⌊ni⌋\lfloor \dfrac ni\rfloor⌊in⌋的值都为⌊nl⌋\lfloor \dfrac nl\rfloor⌊ln⌋。令d=⌊nl⌋d=\lfloor \dfrac nl\rfloord=⌊ln⌋,那么有
d=⌊ni⌋=n−n%iii×d=n−n%i≤ni≤ndd=\lfloor \dfrac ni\rfloor=\dfrac{n-n\%i}{i} \\ \qquad \\i\times d=n-n\% i\leq n \\ \qquad \\i\leq \dfrac ndd=⌊in⌋=in−n%ii×d=n−n%i≤ni≤dn
因为rrr为右端点,所以rrr是最大的,r=⌊nd⌋=⌊n⌊nl⌋⌋r=\lfloor\dfrac nd\rfloor=\lfloor\dfrac{n}{\lfloor \frac nl\rfloor}\rfloorr=⌊dn⌋=⌊⌊ln⌋n⌋
由上文,⌊ni⌋\lfloor \dfrac ni\rfloor⌊in⌋最多只能取2n2\sqrt n2n个值,所以时间复杂度为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=1min(n,m)⌊ni⌋−⌊mi⌋\sum\limits_{i=1}^{\min(n,m)}\lfloor \dfrac ni\rfloor -\lfloor\dfrac mi\rfloori=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,gf,gf,g为数论函数,FFF为fff的前缀和。若F,gF,gF,g已知,则如下式子也可以用数论分块来求。
∑i=1nf(i)×g(⌊ni⌋)\sum\limits_{i=1}^nf(i)\times g(\lfloor\dfrac ni\rfloor)i=1∑nf(i)×g(⌊in⌋)
对于每一块,设左右端点分别为l,rl,rl,r,则这一块的贡献为
(F(r)−F(l−1))×g(⌊nl⌋)(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)。
3435

被折叠的 条评论
为什么被折叠?



