其他分享
首页 > 其他分享> > NTT

NTT

作者:互联网

来,背

稍微提一下原根的求法,我们对于模数(p-1)质因数分解,然后暴力枚举原根,我们发现原根满足 \(^{n}_{i=1}g^{\frac{p-1}{p_i}}\not\equiv 1(mod\ p)\) 就没了。

然后就是模数需要满足是\(X\times 2_n+1\) 的形式,否则上任意模数NTT

#include<bits/stdc++.h>
using namespace std;
#define LL long long 
#define file(a) freopen(#a".in","r",stdin),freopen(#a".out","w",stdout)
#define endless() fclose(stdin),fclose(stdout)
#define ReadOnly(a) freopen(#a,"r",stdin)
inline int read(){
	int s=0,f=1;char ch=getchar();
	while(ch<'0'||'9'<ch) {if(ch=='-') f=-1;ch=getchar();}
	while('0'<=ch&&ch<='9') {s=s*10+(ch^48);ch=getchar();}
	return s*f;
}
const int mods=998244353;
const int N=1e6+5e5+3;
inline int inc(int x,int y){
	return (x+=y)>=mods?x-mods:x;
}
inline int dec(int x,int y){
	return (x-=y)<0?x+mods:x;
}
inline int mul(int x,int y){
	return 1ll*x*y%mods;
}
int qpow(int a,int b){
	int ans=1;
	while(b){
		if(b&1) ans=mul(ans,a);
		a=mul(a,a);
		b>>=1;
	}
	return ans;
}
inline int inv(int x){
	return qpow(x,mods-2);
}
int F[N<<1],P[N<<1];
int rev[N<<1];int n,m;
int G=3;
int w[N<<1];
int invG=inv(G);
inline void get_rev(){
	for(int i=0;i<n;++i){
		rev[i]=(rev[i>>1]>>1)|((i&1)?n>>1:0);
	}
}
inline void NTT(int *f,bool flag){
	for(int i=0;i<n;++i){
		if(i<rev[i]) swap(f[i],f[rev[i]]);
	}
	for(int p=2;p<=n;p<<=1){//枚举区间长度
		int len=p>>1;//获取分治长度
		w[0]=1,w[1]=qpow(flag?G:invG,(mods-1)/p);
		for(int i=2;i<=p;++i){
			w[i]=mul(w[i-1],w[1]);
		}
		for(int k=0;k<n;k+=p){//枚举初始化位置
			int pos=0;
			for(int l=k;l<k+len;++l){//枚举detail
				int tmp=mul(w[pos],f[len+l]);
				f[len+l]=dec(f[l],tmp);
				f[l]=inc(f[l],tmp);
				++pos;//获取下一个单位根
			}
		}
	}
}
int main(void){
	n=read();m=read();
	for(int i=0;i<=n;++i){
		F[i]=read();
	}
	for(int i=0;i<=m;++i){
		P[i]=read();
	}
	for(m+=n,n=1;n<=m;n<<=1);//补足2的幂次
	get_rev();//二进制翻转
	NTT(F,1);NTT(P,1);
	for(int i=0;i<n;++i) F[i]=mul(F[i],P[i]);
	NTT(F,0);
	for(int i=0;i<=m;++i){
		printf("%d ",(int)mul(F[i],inv(n)) );
	}
	return 0;
}

求原根

#include<bits/stdc++.h>
#define ll long long
#define db double 
#define filein(a) freopen(#a".in","r",stdin)
#define fileot(a) freopen(#a".out","w",stdout)
using std::mt19937;
using std::sort;
namespace TOOLS{
	const db eps=1e-9;
	template<class T>
	inline void read(T &s){
		s=0;bool f=0;char ch=getchar();
		while(ch<'0'||'9'<ch) {if(ch=='-') f=1;ch=getchar();}
		while('0'<=ch&&ch<='9') {s=s*10+(ch^48);ch=getchar();}
		if(ch=='.'){
			T p=1;ch=getchar();
			while('0'<=ch&&ch<='9') {p*=0.1;s=s+(ch^48)*p;ch=getchar();}
		}
		s=f?-s:s;
	}
	template<class T>
	inline int sgn(T x){
		return x<-eps?-1:x<eps?0:1;
	}
	mt19937 rd_ori(time(0)+114514*time(0) );
	template<class T>
	inline T rd(T L,T R){
		return std::uniform_int_distribution<T>(L,R)(rd_ori);
	}
}
using namespace TOOLS;
const int N=100+3;
int mods;
inline int inc(int x,int y){
	return (x+=y)>=mods?x-mods:x;
}
inline int dec(int x,int y){
	return (x-=y)<0?x+mods:x;
}
inline int mul(int x,int y){
	return 1ll*x*y%mods;
}
inline int qpow(int a,int b){
	int ans=1;
	while(b){
		if(b&1) ans=mul(ans,a);
		a=mul(a,a);
		b>>=1;
	}
	return ans;
}
inline int inv(int x) {return qpow(x,mods-2);}
int pri[N];
int main(){
	//filein(a);//fileot(a);
	read(mods);
	int x=mods-1;
	for(int i=2;i*i<=x;++i){
		if(x%i==0){
			pri[++pri[0] ]=i;
			while(x%i==0) x/=i;
		}
	}
	if(x!=1){
		pri[++pri[0] ]=x;
	}
	for(int g=2;g<=mods;++g){
		bool flag=0;
		for(int i=1;i<=pri[0];++i){
			if(qpow(g,mul(mods-1,inv(pri[i]) ) )==1){
				flag=1;break;
			}
		}
		if(!flag){
			printf("%d\n",g);
			break;
		}
	}
	return 0;
}

标签:mods,return,int,NTT,freopen,inline,define
来源: https://www.cnblogs.com/cbdsopa/p/15930374.html