之前做题要用到min25,就断断续续地学了几下,用后即忘,简直就是浪费时间。不如现在好好记下来,巩固一下记忆。
在找博客学习过程中发现了一个写得非常好的博客:Min-25筛学习笔记 | LNRBHAW,配合Min_25 筛 - OI Wiki (oi-wiki.org)食用,效果很好。
min25可以以低于线性的复杂度计算出一类积性函数\(f\)的前缀和(不一定是局限于积性函数,见题目Hasse Diagram)。
要求对于素数\(p\),\(f(p)\)是一个关于\(p\)的低次多项式,并且\(f(p^c)\)可以快速计算出来。例如\(f(p)=p^2-p\),\(f(p^k)=p^k(p^k-1)\)这样的函数。
定义:
\({\rm lpf}(n)\)为\(n\)的最小质因数。例如\({\rm lpf}(6)=2\),特别地,\({\rm lpf}(1)=1\)。
\(p_k\)代表第\(k\)大的质数。
\(S(n,k)=\sum\limits_{i=2}^{n}{[p_k<{\rm lpf}(i)]f(i)}\)
\(F_{\rm prime}(n)=\sum\limits_{p_i\le n}{f(p_i)}\)
这样,\(S(n,1)+f(1)\)就是我们要求的前缀和。
\(S(n,k)\)中可以分为两个部分分别计算,一个是素数的部分,一个合数部分。
素数:即为\(S(n,k)=F_{\rm prime}(n)-F_{\rm prime}(p_{k-1})\)
合数:通过枚举每个\(i\)的最小质因子及其次数就可以求得递推式(枚举最小质因子保证了不重不漏)。
因为\(f(p)=\sum{a_kp^k}\),有
\(F_{\rm prime}(n)=\sum\limits_{p_i\le n}{f(p_i)}=\sum\limits_{p_i\le n}{\sum{a_kp_i^k}}=\sum{a_k\sum\limits_{p_i\le n}{p_i^k}}=\sum{a_k\sum\limits_{p_i\le n}{g(p_i)}}\)
令\(g(p)=p^k\),可以发现\(g(p)\)是完全积性函数(注意\(g(p)\)系数是1,否则就不是完全积性函数了)。所以现在要求的是\(F_{\rm prime}(n)=\sum\limits_{p_i\le n}{g(p_i)}\)
设函数\(G(n,i)\)为用前\(i\)个质数做埃式筛剩下的\(g\)值之和,这样\(F_{\rm prime}(n)=G(n,\lfloor\sqrt{n}\rfloor)\)
它由有两个部分组成:
\(p_i^2>n\),筛完了,\(G(n,i)=G(n,i-1)\)
\(p_i^2\le n\),\(p_i\)的在区间\([p_i,\lfloor \frac{n}{p_i}\rfloor]\)且没被前\(i-1\)个素数筛掉的倍数会被筛掉,要减去。而这些倍数位于\(G(\lfloor \frac{n}{p_i}\rfloor,i-1)\)中,除此之外,\(G(\lfloor \frac{n}{p_i}\rfloor,i-1)\)还包含了\(p_1,p_2,...,p_{i-1}\)这些值没被筛去,要从中减去\(F_{\rm prime}(p_{i-1})\)
综上有
\(G(n,i)=G(n,i-1)-[p_i^2\le n]g(p_i)(G(\lfloor \frac{n}{p_i}\rfloor,i-1)-F_{\rm prime}(p_{i-1}))\)
注意\(g(p)=p^k,p{\rm 是质数}\),是完全积性函数,故解析式为\(g(x)=x^k,x\in Z^+\),可以O(1)求出其\(g\)前缀和\(G(n,0)\),\(F_{\rm prime}(p_{i-1})\)也可以在O(\(\sqrt{n}\))预处理出来。
可以发现都是形如\(\lfloor \frac{n}{p_i}\rfloor\),可以使用数论分块+递推预处理出\(2\sqrt{n}\)项\(F_{\rm prime}\)值,分别为\(F_{\rm prime}(i),i=1,...,\sqrt{n}\)和\(F_{\rm prime}(\frac{n}{i}),i=1,...,\sqrt{n}\)。这\(2\sqrt{n}\)个就是下面要用的值。
枚举最小质数可得递推式如下:
\(S(n, k)=\sum\limits_{i\ge k}\sum\limits_{c>1}^{p_i^{c+1}\le n}{(f(p_i^c)S(\frac{n}{p_i^c},i+1)+f(p_i^{c+1})})\)
由于\(f(x)\)是积性函数,\(f(p_i^c)S(\frac{n}{p_i^c},i+1)\)相当于最小质因数为\(p_i^c\)的合数对应的值的和,最后还要补上一个\(f(p_i^{c+1})\)。
最终结果即为:
\(S(n, k)=\sum\limits_{i\ge k}\sum\limits_{c>1}^{p_i^{c+1}\le n}{(f(p_i^c)S(\frac{n}{p_i^c},i+1)+f(p_i^{c+1})})+F_{\rm prime}(n)-F_{\rm prime}(p_{k-1})\)
直接递归计算即可。
实现中将\(F_{\rm prime}(i),i=1,...,\sqrt{n}\)这小于\(\sqrt{n}\)的映射到\(\rm id1[i]\),\(F_{\rm prime}(\frac{n}{i}),i=1,...,\sqrt{n}\)这些大于\(n\)的映射到\(\rm id2[i]\),相当于hashmap。
求素数部分用的是非递归写法,直接递推上去即可。
P5325 【模板】Min_25筛
#include <bits/stdc++.h> using namespace std; const int N = 1e6 + 10; const int M = 1e9 + 7; const int INF = 0x3f3f3f3f; typedef long long ll; ll num[N], ans1[N], ans2[N]; ll id1[N], id2[N]; ll pri[N]; ll m, n; int cnt; bool isnp[N]; void init() { isnp[1] = 1; pri[cnt++] = 1; for(int i = 2; i < N; i++) { if(!isnp[i]) { pri[cnt++] = i; for(int j = 2 * i; j < N; j += i) isnp[j] = 1; } } } ll r2 = 500000004; ll r6 = 166666668; ll sum1(ll x) { // 求和,用于求初始值S(n,0) x %= M; return x * (x + 1) % M * r2 % M; } ll sum2(ll x) { x %= M; return x * (x + 1) % M * (2 * x + 1) % M * r6 % M; } ll f1(ll x) { // 一次幂 x %= M; return x; } ll f2(ll x) { // 二次幂 x %= M; return x * x % M; } ll f(ll x) { // 真正的 x %= M; return x * (x - 1) % M; } int ID(ll x) { // hash映射 if(x <= m) return id1[x]; return id2[n / x]; } void solve(ll n, ll (*f)(ll), ll (*sum)(ll), ll ans[]) { static ll sump[N]; // 预处理素数和 int cur = 0; int mx = 0; m = 0; while(m * m <= n) { if(pri[mx] <= m) { mx++; sump[mx] = (sump[mx - 1] + f(pri[mx])) % M; } m++; } ll i = 1; while(i <= n) { // 数论分快求出映射关系和初始值 num[cur] = n / i; ans[cur] = (sum(num[cur]) - f(1)) % M; if(num[cur] <= m) id1[num[cur]] = cur; else id2[n / num[cur]] = cur; cur++; i = n / (n / i) + 1; } for(int i = 1; i < mx; i++) { // 直接递推 for(int j = 0; j < cur && 1ll * pri[i] * pri[i] <= num[j]; j++) { ans[j] -= f(pri[i]) * (ans[ID(num[j] / pri[i])] - sump[i - 1]) % M; ans[j] %= M; } } } ll F(ll n, int k) { ll res = 0; for(int i = k; 1ll * pri[i] * pri[i] <= n; i++) { ll p = pri[i]; while(p * pri[i] <= n) { res += (f(p) * F(n / p, i + 1) % M + f(p * pri[i])) % M; res %= M; p = p * pri[i]; } } res += (ans2[ID(n)] - ans1[ID(n)]) - ((k - 1) ? (ans2[ID(pri[k - 1])] - ans1[ID(pri[k - 1])]) : 0) % M; return res % M; } int main() { init(); while(cin >> n) { solve(n, f1, sum1, ans1); // 素数部分 solve(n, f2, sum2, ans2); ll ans = (F(n, 1) + 1 + M) % M; // 合数+素数统计 cout << (ans % M + M) % M << endl; } }