题解 P5325 【【模板】Min_25筛】

wucstdio

2019-04-27 21:53:36

题解

洛谷题库里面终于有我的题了(虽然是一道板子),必须发一份题解庆祝一下。

题意

\sum_{i=1}^nf(i),其中f(i)是一个积性函数,且f(p^k)=p^k(p^k-1)

Min_25筛

为什么这个筛法叫做Min25筛呢?因为这个算法是Min25发明的。

假设我们要求一个\sum_{i=1}^nf(i),满足f(x)是一个积性函数,且f(p^e)是一个关于p的低阶多项式。

因为多项式可以拆成若干个单项式,所以我们只需要考虑求出f(p)=p^k的前缀和,然后每一项加起来就行了。

那么如何求出每一项的和呢?

Step1 分类

让我们先对i按照质数和合数分类:

\sum_{i=1}^nf(i)=\sum_{1\le p\le n}f(p)+\sum_{i=1\&\text{i is not a prime}}^nf(i)

然后,我们枚举后面合数的最小质因子以及最小质因子的次数。注意所有合数的最小质因子一定都小于等于\sqrt n

\sum_{1\le p\le n}f(p)+\sum_{1\le p^e\le n,1\le p\le \sqrt n}f(p^e)\left(\sum_{1\le i\le n/p^e\&minp>p}f(i)\right)

其中minp表示i的最小质因子,因为公式中文太丑了所以就只好写英文了。

这样,整个式子就变成了两个部分,第一部分是所有质数的f之和,另一部分是枚举最小质因子后,求所有最小质因子大于这个质因子的f之和。

Step2 第一部分

由于n实在太大,所以我们没法用线性筛求出所有质数的f和。

我们考虑一个DP的思路(天哪这是怎么想到的):我们不知道从哪里找来了一个DP数组g(n,i),满足

g(n,j)=\sum_{i=1}^n[\text{i is a prime or minp$> p_j$}]i^k

这里的k就是前面我们说的低阶多项式的一项。注意i^k并不是我们要求的f,只是一个和f在质数处的取值一样的完全积性函数,这样后面计算起来比较方便。

这个式子说人话就是g(n,j)表示求1n之间所有满足条件的数的k次方和,条件就是要么是质数要么最小质因子大于p_j

我们考虑g(n,j-1)如何转移到g(n,j)。随着j的增大,满足条件的数变少了,所以我们需要减去一些原来满足条件而现在不满足条件的数。

这些数应该是最小质因子恰好为p_{j}的合数。

我们可以提出来一个p_{j}作为最小质因子,这样剩下的就不能有小于它的质因子了,也就是g\left(\dfrac{n}{p_{j}},j\right)-g(p_{j-1},j-1),后面那个g是为了把所有的质数去掉。

这样我们就得到了g的递推式:

g(n,j)=g(n,j-1)-p_j^k\left(g\left(\dfrac{n}{p_j},j-1\right)-g(p_{j-1},j-1)\right)

完全积性函数的好处在这里就体现出来了:由于只提出了一个p_i,所以后面还有可能有p_i这个因子,如果是完全积性函数的话就可以将函数值直接相乘,而不用管是否互质。

注意到后面的g(p_{j-1},j-1)其实就是前j-1个质数的k次方和。由于p_j\le \sqrt n,所以这一部分可以用线性筛预处理,我们设sp_n=\sum_{i=1}^np_i^k,也就是前n个质数的k次方和。

但是因为$n$太大,我们还是没法对于每一个$n$求出$g(n,x)$,所以我们可以想到另一个重要的结论: $$\left\lfloor\dfrac{\lfloor\dfrac na\rfloor}{b}\right\rfloor=\lfloor\dfrac{n}{ab}\rfloor$$ 也就是说,无论你每一次把$n$除以几,最后你能得到的数一定是某一个$\lfloor\dfrac nx\rfloor$,所以我们没必要算出来所有的$n$,只需要算出可以写成$\lfloor\dfrac{n}{x}\rfloor$这种形式的数,这样的数一共有$O(\sqrt n)$个。 那么我们如何存储这$\sqrt n$个数呢? 首先我们不能直接下标访问,这样下标可以到$n$。我们需要对下标离散化。 但是离散化之后,我们还需要知道对于每一个$\lfloor\dfrac nx\rfloor$,它对应的下标是什么。 如果偷懒的话可以用map,但是时间复杂度会多一个$\log$。我们可以用$ind1[x]$表示$x$这个数对应的数组下标,$ind2[x]$表示$n/x$这个数对应的下标。这样两个$ind$数组最大都只会到$\sqrt n$。 具体实现可以看代码。数组的记录上需要精细实现一下。 ### Step2 求解答案 答案就是先求出所有质数的函数和,然后先枚举了一个$p^e$,再枚举最小质因子大于$p$的数。 我们还是可以考虑DP的思想。设$S(n,x)$表示求$1$到$n$中所有最小质因子大于$p_x$的函数值之和,注意这里是$f$而不是$k$次方。答案就是$S(n,0)$。 我们将满足条件的数分成两部分,第一部分是大于$p_x$的质数,也就是$g(n)-sp_x$,另一部分是最小质因子大于$p_x$的合数,枚举最小质因子: $$S(n,x)=g(n)-sp_x+\sum_{p_k^e\le n\&k>x}f(p_k^e)\left(S\left(\dfrac{n}{p_k^e},k\right)+[e\neq 1]\right)$$ 这样问题就解决了,我们可以递归求解这个问题。根据某玄学定理,不需要记忆化。 ### 一些细节 $1$既不是质数也不是合数,不含任何一个质因子,那么求解的过程中$g$和$S$到底是否包含$1$呢?其实是否包含都可以,但是处理上略有差别。我的$g$和$S$都没有包含$1$,只需要最后加一就行了。 min25筛的时间复杂度据说是$O\left(\dfrac{n^{3/4}}{\log n}\right)$,也有人说是$O(n^{1-\epsilon})$,在这道题上大致是1e10跑1s左右的样子。但是这个算法常数很小,具体表现参加WC2019课件里面的一张图: ![](https://cdn.luogu.com.cn/upload/pic/56355.png) ## 代码实现 ```cpp #include<cstdio> #include<algorithm> #include<cstring> #include<cmath> #define ll long long using namespace std; const ll MOD=1000000007,inv2=500000004,inv3=333333336; ll prime[1000005],num,sp1[1000005],sp2[1000005]; ll n,Sqr,tot,g1[1000005],g2[1000005],w[1000005],ind1[1000005],ind2[1000005]; bool flag[1000005]; void pre(int n)//预处理,线性筛 { flag[1]=1; for(int i=1;i<=n;i++) { if(!flag[i]) { prime[++num]=i; sp1[num]=(sp1[num-1]+i)%MOD; sp2[num]=(sp2[num-1]+1ll*i*i)%MOD; } for(int j=1;j<=num&&prime[j]*i<=n;j++) { flag[i*prime[j]]=1; if(i%prime[j]==0)break; } } } ll S(ll x,int y)//第二部分 { if(prime[y]>=x)return 0; ll k=x<=Sqr?ind1[x]:ind2[n/x]; ll ans=(g2[k]-g1[k]+MOD-(sp2[y]-sp1[y])+MOD)%MOD; for(int i=y+1;i<=num&&prime[i]*prime[i]<=x;i++) { ll pe=prime[i]; for(int e=1;pe<=x;e++,pe=pe*prime[i]) { ll xx=pe%MOD; ans=(ans+xx*(xx-1)%MOD*(S(x/pe,i)+(e!=1)))%MOD; } } return ans%MOD; } int main() { scanf("%lld",&n); Sqr=sqrt(n); pre(Sqr); for(ll i=1;i<=n;) { ll j=n/(n/i); w[++tot]=n/i; g1[tot]=w[tot]%MOD; g2[tot]=g1[tot]*(g1[tot]+1)/2%MOD*(2*g1[tot]+1)%MOD*inv3%MOD; g2[tot]--; g1[tot]=g1[tot]*(g1[tot]+1)/2%MOD-1; if(n/i<=Sqr)ind1[n/i]=tot; else ind2[n/(n/i)]=tot; i=j+1; }//g1,g2分别表示一次项和二次项,ind1和ind2用来记录这个数在数组中的位置 for(int i=1;i<=num;i++)//由于g数组可以滚动,所以就只开了一维 { for(int j=1;j<=tot&&prime[i]*prime[i]<=w[j];j++) { ll k=w[j]/prime[i]<=Sqr?ind1[w[j]/prime[i]]:ind2[n/(w[j]/prime[i])]; g1[j]-=prime[i]*(g1[k]-sp1[i-1]+MOD)%MOD; g2[j]-=prime[i]*prime[i]%MOD*(g2[k]-sp2[i-1]+MOD)%MOD; g1[j]%=MOD,g2[j]%=MOD; if(g1[j]<0)g1[j]+=MOD; if(g2[j]<0)g2[j]+=MOD; } } printf("%lld\n",(S(n,0)+1)%MOD); return 0; } ```