【洛谷】简单的数学题-莫比乌斯反演&杜教筛

传送门:luoguP3768


题解

这个方法比较复杂但是很有技巧性。

这个方法比较直接简洁。

(懒


代码

#include
using namespace std;
const int N=8e6;
typedef long long ll;

ll n;int mod,iv2,iv6;
int p[N],tot,phi[N],ans,lim;
bool pri[N];
map<ll,int>mp;

inline int fp(int x,int y)
{
    int re=1;
    for(;y;y>>=1,x=(ll)x*x%mod) if(y&1) re=(ll)re*x%mod;
    return re;
}

inline int ad(int x,int y){return ((ll)x+y)%mod;}
inline int dc(int x,int y){return ((ll)x-y+mod)%mod;}

inline void pre()
{
    lim=(n<N)?(int)n:(N-1);int re,i,j;phi[1]=1;
    for(int i=2;i<=lim;++i){
        if(!pri[i]) p[++tot]=i,phi[i]=i-1;
        for(j=1;j<=tot && (re=i*p[j])<=lim;++j){
            pri[re]=true;
            if(i%p[j]==0) {phi[re]=(ll)phi[i]*p[j]%mod;break;}
            phi[re]=(ll)phi[i]*phi[p[j]]%mod;
        }
    }
    for(i=2;i<=lim;++i) phi[i]=ad(phi[i-1],(ll)i*i%mod*(ll)phi[i]%mod);
}

inline int S(int x){return (ll)x*(x+1)%mod*(ll)iv2%mod;}
inline int Sp(int x){return (ll)x*(x+1)%mod*(ll)(((ll)x+x+1)%mod)%mod*(ll)iv6%mod;}

int F(ll x)
{
    if(x<=lim) return phi[x];
    if(mp[x]) return mp[x];
    int re=S(x%mod);re=(ll)re*re%mod;
    for(ll j,i=2;i<=x;i=j+1){
        j=x/(x/i);
        re=dc(re,(ll)dc(Sp(j%mod),Sp((i-1)%mod))*F(x/i)%mod);
    }
    return (mp[x]=re);
}

int main(){
    ll i,j;int ss,vl,pr=0;
    scanf("%d%lld",&mod,&n);pre();
    iv2=fp(2,mod-2);iv6=fp(6,mod-2);
    for(i=1;i<=n;i=j+1,pr=vl){
        j=n/(n/i);ss=S((n/i)%mod);ss=(ll)ss*ss%mod;
        vl=F(j);ans=ad(ans,(ll)ss*dc(vl,pr)%mod);
    }
    printf("%d",ans);
    return 0;
}

你可能感兴趣的:(积性函数)