洛谷P4213 【模板】杜教筛（Sum）

题目链接

杜教筛模板题

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int N=4e6+10;
int phi[N],pri[N],np,isp[N],mu[N];
ll sphi[N],smu[N];
unordered_map<int,ll> mpsphi,mpsmu;
ll Sphi(int n) {
    if(n<N)return sphi[n];
    if(mpsphi.count(n))return mpsphi[n];
    ll& ret=mpsphi[n];
    ret=(ll)n*(n+1)/2;
    for(int l=2,r,k; l<=n; l=r+1) {
        k=n/l,r=n/k;
        ret-=Sphi(n/l)*(r-l+1);
    }
    return ret;
}
ll Smu(int n) {
    if(n<N)return smu[n];
    if(mpsmu.count(n))return mpsmu[n];
    ll& ret=mpsmu[n];
    ret=1;
    for(int l=2,r,k; l<=n; l=r+1) {
        k=n/l,r=n/k;
        ret-=Smu(n/l)*(r-l+1);
    }
    return ret;
}
int main() {
    for(int i=2; i<N; ++i)isp[i]=1;
    phi[1]=mu[1]=1;
    for(int i=2; i<N; ++i) {
        if(isp[i]) {pri[np++]=i,phi[i]=i-1,mu[i]=-1;}
        for(int j=0; j<np&&i*pri[j]<N; ++j) {
            isp[i*pri[j]]=0;
            if(i%pri[j]==0) {
                phi[i*pri[j]]=phi[i]*pri[j];
                mu[i*pri[j]]=0;
                break;
            }
            phi[i*pri[j]]=phi[i]*(pri[j]-1);
            mu[i*pri[j]]=-mu[i];
        }
    }
    for(int i=1; i<N; ++i)sphi[i]=sphi[i-1]+phi[i];
    for(int i=1; i<N; ++i)smu[i]=smu[i-1]+mu[i];
    int T;
    for(scanf("%d",&T); T--;) {
        int n;
        scanf("%d",&n);
        printf("%lld %lld
",Sphi(n),Smu(n));
    }
    return 0;
}