【HDU5608】function（莫比乌斯反演）（杜教筛）

最新推荐文章于 2021-06-13 13:56:37 发布

原创最新推荐文章于 2021-06-13 13:56:37 发布 · 227 阅读

1 ·

本内容遵循CC 4.0 BY-SA版权协议

筛法同时被 2 个专栏收录

26 篇文章

订阅专栏

莫比乌斯反演

24 篇文章

订阅专栏

本文深入解析了杜教筛算法与莫比乌斯反演在解决复杂数学问题中的应用，通过预处理调和级数及组合数，有效提升了计算效率。文章详细介绍了算法实现过程，并附带了C++代码示例。

传送门

题解：

先莫比乌斯反演+调和级数预处理出部分前缀和。

杜教筛的式子非常显然：

$S(n)={n\choose 3}-\sum_{i=2}^nS(\lfloor\frac{n}{i}\rfloor)$

代码：

#include<bits/stdc++.h>
#define ll long long
#define re register
#define cs const

using std::cerr;
using std::cout;

cs int mod=1e9+7;
inline int add(int a,int b){return a+b>=mod?a+b-mod:a+b;}
inline int dec(int a,int b){return a-b<0?a-b+mod:a-b;}
inline int mul(int a,int b){ll r=(ll)a*b;return r>=mod?r%mod:r;}
inline void Inc(int &a,int b){a+=b-mod;a+=a>>31&mod;}
inline void Dec(int &a,int b){a-=b;a+=a>>31&mod;}
inline void Mul(int &a,int b){a=mul(a,b);}
cs int iv3=(mod+1)/3;

cs int N=1e6+7;

int s[N],n;

void init(){
	for(int re i=1;i<N;++i)
		s[i]=mul(i-1,i-2);
	for(int re i=1;i<N;++i)
		for(int re j=i+i;j<N;j+=i)
			Dec(s[j],s[i]);
	for(int re i=2;i<N;++i)
		Inc(s[i],s[i-1]);
}

std::unordered_map<int,int> Map;

int S(int n){
	if(n<N)return s[n];
	if(Map.find(n)!=Map.end())return Map[n];
	int res=mul(mul(n,iv3),mul(n-1,n-2));
	for(int re l=2,r,u;(u=n/l);l=r+1)
		Dec(res,mul((r=n/u)-l+1,S(u)));
	return Map[n]=res;
}

void Main(){
	init();int T;
	scanf("%d",&T);
	while(T--){
		scanf("%d",&n);
		cout<<S(n)<<"\n";
	}
}

inline void file(){
#ifdef zxyoi
	freopen("function.in","r",stdin);
#endif
}
signed main(){file();Main();return 0;}