题意:
杜教筛模版题,求欧拉函数的前缀和,以及莫比乌斯函数的前缀和
杜教筛知识详解
代码:
#include <iostream>
#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
const int maxn=2e6+5;
ll sum[maxn];//莫比乌斯函数的前缀和
int prime[maxn],mo[maxn];//素数,莫比乌斯函数
bool vis[maxn];
ll psum[maxn],phi[maxn];//欧拉函数的前缀和,欧拉函数的值
//为了更省空间,可以在phi,mo数组上直接累加,不用开sum,psum数组,如果maxn开到5e6,预处理的更多,时间会更快
map<int,ll>p;
map<int,ll>p2;
int cnt;
void get(){
mo[1]=1,phi[1]=1;
for(int i=2;i<=maxn;i++){
if(!vis[i]){
prime[++cnt]=i;//记录素数
mo[i]=-1;
phi[i]=i-1;
}
for(int j=1;j<=cnt&&i*prime[j]<=maxn;j++){
vis[i*prime[j]]=1;
if(i%prime[j]==0){
phi[i*prime[j]]=phi[i]*prime[j];
break;
}
mo[i*prime[j]]=-mo[i];
phi[i*prime[j]]=phi[i]*phi[prime[j]];
}
}
for(int i=1;i<=maxn;i++)sum[i]=(sum[i-1]+mo[i]),psum[i]=psum[i-1]+phi[i];
}
ll dujiu(ll n){//莫比乌斯函数
if(n<=maxn)return sum[n];
ll ans=1;
if(p[n])return p[n];
for(ll l=2,r;l<=n;l=r+1){
r=n/(n/l);
ans=(ans-(r-l+1)*dujiu(n/l));
}
return p[n]=ans;
}
ll dujip(ll n){//欧拉函数
if(n<=maxn)return psum[n];
ll ans=n*(n+1)/2;
if(p2[n])return p2[n];
for(ll l=2,r;l<=n;l=r+1){
r=n/(n/l);
ans=(ans-(r-l+1)*dujip(n/l));
}
return p2[n]=ans;
}
int main()
{
int n;
get();
int t;
scanf("%d",&t);
while(t--){
scanf("%d",&n);
printf("%lld %lld\n",dujip(n),dujiu(n));
}
//cout << "Hello world!" << endl;
return 0;
}