杜教筛模板

杜教筛模板

#include<cstdio>
#include<algorithm>
#include<cstring>
#include<map>
using namespace std;
typedef long long LL;
const int N=2e6+10;
//primes[i] 存素数
//mu[i]  存i的莫比乌斯函数值
//sum_mu[i] 存1~n 的莫比乌斯函数值的前缀和 先用线性筛筛出前面一部分 
LL primes[N],mu[N],sum_mu[N],cnt,T,n;
bool st[N];
//莫比乌斯函数的map 防止爆空间
map<LL,LL> mp_mu;
LL s_mu(LL x){
	//在线性筛范围内
	if(x<N){
		return sum_mu[x];
	}
	//图中已经存了 
	if(mp_mu[x]){
		return mp_mu[x]; 
	}
	LL res=1LL;
  	for(LL i=2,j;i<=x;i=j+1){
    	j=x/(x/i);
    	res-=s_mu(x/i)*(j-i+1);
  	}
  	return mp_mu[x]=res;
}
//求欧拉函数前缀和 
LL s_phi(LL x){
	LL res=0LL;
	for(LL i=1,j;i<=x;i=j+1){
		j=x/(x/i);
		res+=(s_mu(j)-s_mu(i-1))*(x/i)*(x/i);
	}
	return ((res-1)>>1)+1;
}
int main(){
	scanf("%lld",&T);
	//线性筛求素数同时维护前一部分莫比乌斯函数值
	mu[1]=1;
	for(int i=2;i<N;i++){
		if(!st[i]){
			//使得下标从1开始 方便维护前缀和 
			primes[++cnt]=i;
			//质数的莫比乌斯函数值都为-1 
			mu[i]=-1;
		}
		for(int j=1;j<=cnt && primes[j]*i<N;j++){
			st[primes[j]*i]=true;
			if(i%primes[j]){
				//i%primes[j]!=0 则与primes[j]*i 本质不同的约数个数比i多一个 
				mu[primes[j]*i]=-mu[i];
			}
			else{
				//i%primes[j]==0 则primes[j]*i%primes[j]^2==0 
				mu[primes[j]*i]=0;
				break;
			}
		}		
	}
	//求前一部分前缀和 
	for(int i=1;i<N;i++){
		sum_mu[i]=sum_mu[i-1]+mu[i];
	}
	while(T--){
		scanf("%lld",&n);
		printf("%lld %lld\n",s_phi(n),s_mu(n));
	}
	return 0;
}