牛客练习赛69 F.解方程（贝尔级数+线性筛）

最新推荐文章于 2022-07-19 14:30:00 发布

m0_52639539

最新推荐文章于 2022-07-19 14:30:00 发布

阅读量174

点赞数 1

分类专栏：数论文章标签： acm竞赛

本文链接：https://blog.csdn.net/m0_52639539/article/details/117307379

版权

数论专栏收录该内容

26 篇文章 0 订阅

订阅专栏

本篇博客主要介绍了牛客练习赛69 F题的解题思路，利用贝尔级数和线性筛解决积性函数问题。通过对σp(n)和σq(n)的分析，导出f(n)的贝尔级数表达式，并计算质数和质数次方处的值，最后采用线性筛进行高效求解，整体时间复杂度为O(n)。

摘要由CSDN通过智能技术生成

牛客练习赛69 F.解方程（贝尔级数+线性筛）

传送门：https://ac.nowcoder.com/acm/contest/7329/F
题目大意：
在这里插入图片描述
题解：不难发现， $f (n)$ 一定为积性函数，这是因为 $\sigma_p(n)$ 和 $\sigma_q(n)$ 均为积性函数，此时便可以使用贝尔级数展开求出 $f (n)$ 的表达式。考虑 $\sigma_k=\operatorname{id}_k*1$ ，则 $\sigma_k(n)$ 的贝尔级数形式就是 ${\sigma_k}_p={\operatorname{id}_k}_p\times 1_p=\frac{1}{(1-x)(1-p^kx)}$ 。为了避免记号混淆，令题目中的 $p, q$ 改为 $a, b$ 。那么：

$f_p\times {\sigma_a}_p={\sigma_b}_p \\f_p\times \frac{1}{(1-x)(1-p^ax)}=\frac{1}{(1-x)(1-p^bx)} \\f_p=\frac{1-p^ax}{1-p^bx} \\f_p=(\operatorname{id}_a\mu)_p\times {\operatorname{id}_b}_p \\f(n)=(\operatorname{id}_a\mu*\operatorname{id}_b)(n)$

计算 $f (n)$ 在质数或质数次方数处的取值，得到 $f(p^r)=p^{rb}-p^{(r-1)b+a}\space (r\geq 1)$ ，不难发现，对于每一个 $f(p^k)\space (k>1)$ ，都有 $f(p^k)=p^bf(p^{k-1})$ ，此时用线性筛维护即可。时间复杂度 $O (n)$ 。

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const ll N=1e7+5,M=998244353;
int cnt=0,prime[N],f[N],bmi[N];
bitset<N> vis;
ll qpow(ll a,ll b){
    ll res=1;
    while(b){
        if(b&1) res=(res*a)%M;
        a=(a*a)%M;
        b>>=1;
    }
    return res;
}
void init(ll n,ll a,ll b){
    bmi[1]=f[1]=1;
    for(int i=2;i<=n;++i){
        if(!vis[i]){
            prime[cnt++]=i;
			bmi[i]=qpow(i,b); 
            f[i]=((ll)bmi[i]-(ll)qpow(i,a))%M;
        }
        for(int j=0;i*prime[j]<=n;++j){
            int t=i*prime[j];
            vis[t]=1;
            if(i%prime[j]){
                f[t]=((ll)f[i]*(ll)f[prime[j]])%M;
            }
            else{
            	f[t]=((ll)f[i]*(ll)bmi[prime[j]])%M;
            	break;
			}
        }
    }
}
int main(){
    ll n,a,b;
    cin>>n>>a>>b;
    init(n,a,b);
    ll res=1;
    for(int i=2;i<=n;++i){
        ll t=f[i];
        if(t<0) t+=M;
        res^=t;
    }
    cout<<res<<endl;
}