杜教筛学习记录

杜教筛

本文参考pengym的杜教筛
杜教筛需要一些预备知识：数论分块(可以参考之前的文章ACM数论初步)、积性函数、狄利克雷卷积，最好了解莫比乌斯函数

积性函数

对于数论函数$f(x)$，如果对于任意满足$\gcd (x,y)=1$的$x,y$，有$f(x\cdot y)=f(x)\cdot f(y)$，则称$f(x)$为积性函数；如果去掉互质的条件仍能满足后面的性质成立，则称$f(x)$为完全积性函数

几个常见的积性函数：

1. $\mu (n)$——莫比乌斯函数，在莫比乌斯反演讲过了
2. $\phi (n)$——欧拉函数，表示$[1,n-1]$中与$n$互质的数的个数($[1,n]$也一样的，反正$n$和$n$也不互质)
3. $d(n)$——约数个数函数，表示$n$的约数个数，$d(n)=\sum _{d|n}1=\sum _{d=1}^n[d|n]$
4. $\sigma (n)$——约数和函数，表示$n$的各个约数之和，$\sigma (n)=\sum _{d|n}d=\sum _{d=1}^n[d|n]\ast d$

几个常见的完全积性函数：

1. $ϵ(n)$——元函数，只有$n=1$时函数值为1，其他情况为0，$ϵ(n)=[n==1]$
2. $I(n)$——恒等函数，函数值恒为1
3. $id(n)$——单位函数，$id(n)=n$

积性函数还有一个重要的性质：积性函数$\ast$积性函数仍为积性函数，这个性质利用积性函数的定义可以很容易地推出

狄利克雷卷积

定义

两个数论函数$f$和$g$的卷积为$(f\ast g)(n)=\sum _{d|n}f(d)\cdot g(\frac{n}{d})$，读作$f$卷$g$，其实读$g$卷$f$也是一样的，因为有定义很容易看出狄利克雷卷积有交换律(本文中的${}^{\prime }{\ast }^{\prime }$均为狄利克雷卷积，${}^{\prime }{\cdot }^{\prime }$为通常的乘法)

性质

1. 交换律：$f\ast g=g\ast f$
2. 结合律：$(f\ast g)\ast h=f\ast (g\ast h)$
3. 分配律：$(f+g)\ast h=f\ast h+g\ast h$，严格来讲是卷积运算对 '+'运算满足分配律

读者们看到$ϵ(n)$可能觉得这种函数啥用？从近世代数的角度看，记所有数论函数的集合为$S$，$S$和狄利克雷卷积$\ast$就构成了一个代数系$(S,\ast )$，而$ϵ(n)$就是这个代数系的单位元，由于狄利克雷卷积$\ast$满足结合律和交换律，所以$(S,\ast )$是个可换半群，再加上单位元，$(S,\ast ,ϵ)$就是一个可换幺半群，至于它是否为一个群在此不作讨论(近世代数的内容看不懂也没关系，不影响后面的内容，笔者也只是刚学一点，而且学得极烂)

狄利克雷卷积可以用来证明一些性质，例如：

• 莫比乌斯函数$\mu$

  莫比乌斯函数有一个重要性质：$\sum _{d|n}\mu (d)=[n==1]$，下面我们用狄利克雷卷积证明莫比乌斯反演：

  ---

  由已知得
  $$F(n)=\sum _{d|n}f(d)，$$ 等价于$F=f\ast I$，两边同时卷积莫比乌斯函数$\mu$，得
  $$F\ast \mu =(f\ast I)\ast \mu$$ 由结合律得
  $$\begin{gathered} F\ast \mu \\ =(f\ast I)\ast \mu \\ =f\ast (I\ast \mu ) \\ =f\ast \sum _{d|n}I(d)\cdot \mu (\frac{n}{d}) \\ =f\ast \sum _{d|n}\mu (\frac{n}{d}) \end{gathered}$$ 再根据$\mu$的性质$\sum _{d|n}\mu (d)=[n==1]$，上式又化为$f\ast ϵ$，就等于$f$，所以就有
  $$f=F\ast \mu =\sum _{d|n}F(d)\cdot \mu (\frac{n}{d})$$

  ---

  采用狄利克雷卷积的证明简单易懂，比之前的和式变换直观许多

• 欧拉函数$\phi$

  欧拉函数有一个重要的性质：$\sum _{d|n}\phi (d)=n$，用它和狄利克雷卷积，我们可以证明莫比乌斯函数的一个性质：$\frac{\phi (n)}{n}=\sum _{d|n}\frac{\mu (d)}{d}$

  ---

  由已知
  $$\sum _{d|n}\phi (d)=n$$ 将它表示为狄利克雷卷积形式：$\phi \ast I=id$，式子两边同时卷$\mu$，得
  $$\begin{gathered} \phi \ast I\ast \mu =id\ast \mu \\ ⟺\phi \ast ϵ=id\ast \mu \\ ⟺\phi =id\ast \mu \end{gathered}$$ 所以$\phi (n)=\sum _{d|n}\mu (d)\cdot \frac{n}{d}$，移项得
  $$\frac{\phi (n)}{n}=\sum _{d|n}\frac{\mu (d)}{d}$$

  ---

杜教筛

掌握了前面的知识以后，就可以开始杜教筛的学习了。首先，我们应该清楚一个问题：杜教筛到底是用来干什么的？杜教筛是以低于线性时间复杂度计算出积性函数前缀和的筛法，需要注意的是它筛的不是$[1,n]$这么大的范围，它只筛$n$，不过能在筛$n$能顺路筛出一些其他的(具体看下文吧

也就是说我们要求$\sum _{i=1}^{n}f(i)$，其中$f(i)$是积性函数，我们姑且记$\sum _{i=1}^{n}f(i)=S(n)$

为了解决上面的问题，我们构造两个积性函数$h$和$g$，使得$h=f\ast g$，下面我们开始求$\sum _{i=1}^{n}h(i)$
$$\begin{array}{rl}\sum _{i=1}^{n}h(i)& =\sum _{i=1}^n\sum _{d|i}g(d)\cdot f(\frac{i}{d})\\ & =\sum _{d=1}^{n}g(d)\cdot \sum _{i=1}^{\lfloor \frac{n}{d}\rfloor }f(i)\\ & =\sum _{d=1}^{n}g(d)\cdot S(\lfloor \frac{n}{d}\rfloor )\end{array}$$接着，将和式的第一项提出来，得到
$$g(1)\cdot S(n)=\sum _{i=1}^{n}h(i)-\sum _{d=2}^{n}g(d)\cdot S(\lfloor \frac{n}{d}\rfloor )$$这样，如果$h(i)$前缀和能很快地求出，对后面的式子利用分块(不过$g(d)$也要能快速求和，且$d$从2开始，不是从1开始)，可以将求$S(n)$的复杂度降到$O(n^{\frac{2}{3}})$。杜教筛的套路就是找到一个$g$，使得$f\ast g=h$的前缀和能简单求出，顺便$g$的求和也简单。注意，杜教筛不像线性筛，比如线性筛筛$\phi 、\mu$这些函数一遍筛就把$[1,n]$的都筛出来，而杜教筛是只筛出$S(n)$，不过从上面的式子很容易就想到递归求解的方式，可以把递归时求出的一些$S(\lfloor \frac{n}{d}\rfloor )$保存下来，这样可以筛得快一些(因为数组没法开到1e9那种级别，可以用个map来存筛出来的$S$)。一般前几项(比如说前1e6项)直接按照线性筛的筛法筛出来(积性函数有个性质——能被$O(n)$筛出来)，然后$O(n)$求一遍前缀和来算；后面的项用上面推导出来的式子，拿杜教筛筛

来点例子：

• 求$S(n)=\sum _{i=1}^n\mu (i)$
  这里的$f=\mu$，现在的目标是找一个$g$，使得$f$卷$g$得到的函数很好求前缀和，顺便$g$也要比较好求和，从上文我们已经得到一个结论：$\mu \ast I=ϵ$，而$ϵ$的前缀和很好求，就是1，所以有
  $$S(n)=1-\sum _{d=2}^{n}S(\lfloor \frac{n}{d}\rfloor )$$

• 求$S(n)=\sum _{i=1}^n\phi (i)$
  由欧拉函数的性质，我们知道$(\phi \ast I)(n)=n$，也就是$\phi \ast I=id$，$id$求和很简单，所以有
  $$S(n)=\sum _{i=1}^{n}i-\sum _{d=2}^{n}S(\lfloor \frac{n}{d}\rfloor )$$

• 求$S(n)=\sum _{i=1}^{n}i\cdot \phi (i)$
  这个式子乍一看难找$g$来和$f$卷，这时候我们就回到狄利克雷卷积的定义，我们要找的是$h=f\ast g$使得$h$的前缀和比较好求，而$(f\ast g)(n)=\sum _{d|n}(d\cdot \phi (d))\cdot g(\frac{n}{d})$。现在的问题就是前面的$d$在这不好求，如果用$g$能约掉就好了，我们发现$g(\frac{n}{d})$的分母有个$d$，那么令$g=id$，那么$h(n)=\sum _{d|n}\phi (d)\cdot n=n\cdot \sum _{d|n}\phi (d)=n^2$，所以有
  $$S(n)=\sum _{i=1}^n{i}^2-\sum _{d=2}^{n}d\cdot S(\lfloor \frac{n}{d}\rfloor )$$ 再根据$\sum _{i=1}^n{i}^2=\frac{n(n+1)(2n+1)}{6}$，就能筛了

板子

• 洛谷P4213 【模板】杜教筛(Sum)

  #include <cstdio>
  #include <map>
  #define N 5000010
  typedef long long ll;
  
  bool check[N];
  int mu[N], sum_mu[N], phi[N], prime[N >> 2], tot;
  ll sum_phi[N];
  std::map<int, int> map_mu;
  std::map<int, ll> map_phi;
  
  void get(int n) {
  	phi[1] = mu[1]= 1; check[1] = true;
  	for(int i = 2; i <= n; i++) {
          if(!check[i]) prime[++tot] = i, mu[i] = -1, phi[i] = i - 1;
          for(int j = 1; j <= tot && i * prime[j] <= n; j++) {
              check[i * prime[j]] = true;
              if(i % prime[j] == 0) {
                  mu[i * prime[j]] = 0;
                  phi[i * prime[j]] = phi[i] * prime[j];
                  break;
              } else {
                  mu[i * prime[j]] = -mu[i];
                  phi[i * prime[j]] = phi[i] * (prime[j] - 1);
              }
          }
  	}
      for(int i = 1; i <= n; i++)
          sum_mu[i] = sum_mu[i - 1] + mu[i], sum_phi[i] = sum_phi[i - 1] + phi[i];
  }
  
  int djmu(int n) {
      if(n < N) return sum_mu[n];
      if(map_mu.count(n)) return map_mu[n];
      int ans = 1;
      for(int l = 2, r = 0; l <= n; l = r + 1) {
          r = n / (n / l);
          ans -= (r - l + 1) * djmu(n / l);
      }
      map_mu.insert(std::make_pair(n, ans));
      return ans;
  }
  
  ll djphi(int n) {
      if(n < N) return sum_phi[n];
      if(map_phi.count(n)) return map_phi[n];
      ll ans = (ll)(n + 1) * n >> 1;
      for(int l = 2, r = 0; l <= n; l = r + 1) {
          r = n / (n / l);
          ans -= (ll)(r - l + 1) * djphi(n / l);
      }
      map_phi.insert(std::make_pair(n, ans));
      return ans;
  }
  
  int main() {
      get(N - 1);
      int t;
      scanf("%d", &t);
      while(t--) {
          int n;
          scanf("%d", &n);
          printf("%lld %d\n", djphi(n), djmu(n));
      }
      return 0;
  }