[BZOJ3944][杜教筛]Sum

最新推荐文章于 2020-09-19 00:25:36 发布

romiqi_new

最新推荐文章于 2020-09-19 00:25:36 发布

阅读量102

点赞数

分类专栏：杜教筛

本文链接：https://blog.csdn.net/qq_43346903/article/details/97137370

版权

杜教筛专栏收录该内容

2 篇文章 0 订阅

订阅专栏

BZOJ3944

杜教筛模板题：
对于 $\sum_{i=1}^n{\mu_i}$ ，有 $\mu*I=\epsilon$ ，由 $\mu$ 的性质或者容斥原理得到
对于 $\sum_{i=1}^n{\phi_i}$ ，有 $\phi*I=Id$ ，将 $\phi_i$ 看做是有多少个分母为 $i$ 的最简真分数，则对于 $1 - n$ 的所有数，都只对一个 $d ∣ n$ 的 $\phi_d$ 有1的贡献，则通过分母类统计得到 $\phi*I=Id$

Code：

#include<bits/stdc++.h>
#include<tr1/unordered_map>
#define ll long long
using namespace std;
inline int read(){
	int res=0,f=1;char ch=getchar();
	while(!isdigit(ch)) {if(ch=='-') f=-f;ch=getchar();}
	while(isdigit(ch)) {res=(res<<1)+(res<<3)+(ch^48);ch=getchar();}
	return res*f;
}
const int P=8e6+7,N=P-7,INF=0x7fffffff;
int pri[P],cnt,mu[P];
bool pt[P];
ll phi[P];
inline void init(){
	phi[1]=mu[1]=1;
	for(int i=2;i<=N;i++){
		if(!pt[i]) pri[++cnt]=i,phi[i]=i-1,mu[i]=-1;
		for(int j=1;j<=cnt && i*pri[j]<=N;j++){
			pt[i*pri[j]]=1;
			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];
		}
		mu[i]+=mu[i-1];
		phi[i]+=phi[i-1];
	}
}

namespace solve_mu{
	tr1::unordered_map<int,int>sum;
	int get(int n){
		if(n<=N) return mu[n];
		if(sum[n]) return sum[n];
		int ans=1;
		for(int i=2,j;j<INF && i<=n;i=j+1){
			j=n/(n/i);
			ans-=(j-i+1)*get(n/i);
		}
		return sum[n]=ans;
	}
}

namespace solve_phi{
	tr1::unordered_map<int,ll>sum;
	ll get(int n){
		if(n<=N) return phi[n];
		if(sum[n]) return sum[n];
		ll ans=n*((ll)n+1)/2;
		for(int i=2,j;j<INF && i<=n;i=j+1){
			j=n/(n/i);
			ans-=(j-i+1)*get(n/i);
		}
		return sum[n]=ans;
	}
}
int main(){
	init();int t=read();
	while(t--){
    	int n=read();
        cout<<solve_phi::get(n)<<" "<<solve_mu::get(n)<<"\n";
	}
    return 0;
}