标签:快速 int inline play lin 题目 cpp 特点 display
很多常用的数论函数都是积性函数,而在题目中,我们常常需要线性(甚至更高)的筛法。
对于积性函数,我们可以在筛素数的基础上稍加修改,即可完成线性筛。
首先,注意到积性函数的特点:
\[
f(xy)=f(x)\times f(y)
\]
而可以线性筛的积性函数,需要知道以下两个式子的快速求法:
\[
f(p)=?\quad f(p^k)=?\\p\in prime
\]
其中, \(f(p)\) 大多是直接定义,\(f(p^k)\) 大多是递归定义。
我们来回忆一下素数筛的过程:
inp[0]=inp[1]=1;
for(int i=2;i<=n;i++){
if(!inp[i]){
prime[++tot]=i;
}
for(int j=1;j<=tot && i*prime[j]<=n;j++){
int tp=prime[j]*i;
inp[tp]=1;
if(i%prime[j]==0){
break;
}
}
}
在线性筛素数的基础上,我们可以进行线性筛的修改。
首先,对于判定的质数 \(p\) ,可以直接给出定义的值。
之后,对于 \(i\%p\neq0\) ,由于 \(i\) 和 \(p\) 互质,可以直接用积性函数性质推得。
然后,对于 \(i\%p == 0\) :
即 \(i\) 内的最小素因子是 \(p\) ,此刻可以将 \(i\) 内的素因子都除掉,然后就可以用积性函数的性质来递推了。为此,我们要记录一个最小质因子的幂次 \(low_i\) 。
那么递推式就可以表示为:\(f(i\times p)=f(i/low_i)\times f(low_i\times p)\) 。
此处还有一个特殊的判定,当 \(i==low_i\) 时,上式相当于没推,所以我们要用 \(f(p^k)\) 的递推来计算。
那么代码如下:
inp[0]=inp[1]=1;
f[1]=1;
for(int i=2;i<=n;i++){
if(!inp[i]){
prime[++tot]=i;
f[i]=对质数的定义式;
low[i]=i;
}
for(int j=1;j<=tot && i*prime[j]<=n;j++){
int tp=prime[j]*i;
inp[tp]=1;
if(i%prime[j]==0){
if(i!=low[i])
f[tp]=f[i/low[i]]*f[low[i]*prime[j]];
else
f[tp]=对p的次幂的定义式;
low[tp]=low[i]*prime[j];
break;
}
f[tp]=f[i]*f[prime[j]];
low[tp]=prime[j];
}
}
缺点很明显,比较耗空间。(但是题目会给够的
当需要线性筛很多个积性函数时,可以同时进行。
这种基于素数筛的线性筛法,有时不止对积性函数有用,对于一些和素数有关的函数也可以筛出,具体在我写的莫比乌斯反演中有例子。
\(\frak by\;thorn\_\)
标签:快速 int inline play lin 题目 cpp 特点 display
原文地址:https://www.cnblogs.com/thornblog/p/12358432.html