洛谷题库里面终于有我的题了(虽然是一道板子),必须发一份题解庆祝一下。
题意
求\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)表示求1到n之间所有满足条件的数的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;
}
```