多项式算法模板

最新推荐文章于 2022-07-07 19:46:15 发布
最新推荐文章于 2022-07-07 19:46:15 发布
阅读量111 收藏 1
点赞数 1
分类专栏： 多项式
版权声明：本文为博主原创文章，遵循 CC 4.0 BY-SA 版权协议，转载请附上原文出处链接和本声明。
本文链接：https://blog.csdn.net/qq_43346903/article/details/97376745
版权

讲解link
附全套模板：
div除法，取模

#include<bits/stdc++.h>
#define poly vector<int>
#define ll long long
#define mod 998244353
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;
}
inline int add(int x,int y){x+=y;if(x>=mod) x-=mod;return x;}
inline int dec(int x,int y){x-=y;if(x<0) x+=mod;return x;}
inline int mul(int x,int y){return 1ll*x*y%mod;}
inline void Mul(int &x,int y){x=1ll*x*y%mod;}
inline void inc(int &x,int y){x+=y;if(x>=mod) x-=mod;}
inline int ksm(int a,int b){int res=1;for(;b;b>>=1,Mul(a,a)) if(b&1) Mul(res,a);return res;}
namespace Ntt{
	const int N=1e6+5;
	int *w[22],rev[N<<2];
	inline void init(int n){for(int i=0;i<n;i++) rev[i]=(rev[i>>1]>>1)|((i&1)*(n>>1));}
	inline void init_w(){
		for(int i=1;i<=21;i++) w[i]=new int[1<<(i-1)];
		int wn=ksm(3,(mod-1)/(1<<21));
		w[21][0]=1;
		for(int i=1;i<(1<<(20));i++) w[21][i]=mul(w[21][i-1],wn);
		for(int i=20;i;i--)
			for(int j=0;j<(1<<(i-1));j++) w[i][j]=w[i+1][j<<1];
	}
	inline void ntt(poly &f,int n,int kd){
		for(int i=0;i<n;i++) if(i>rev[i]) swap(f[i],f[rev[i]]);
		for(int mid=1,l=1;mid<n;mid<<=1,l++){
			for(int i=0;i<n;i+=(mid<<1)){
				for(int j=0,a0,a1;j<mid;j++){
					a0=f[i+j],a1=mul(f[i+j+mid],w[l][j]);
					f[i+j]=add(a0,a1);f[i+j+mid]=dec(a0,a1);
				}
			}
		}
		if(kd==-1 && (reverse(f.begin()+1,f.begin()+n),1))
			for(int inv=ksm(n,mod-2),i=0;i<n;i++) Mul(f[i],inv);
	}	
	inline poly operator -(poly a,poly b){
		poly c;int lim=max(a.size(),b.size());c.resize(lim);
		for(int i=0;i<lim;i++)c[i]=dec(a[i],b[i]);return c;
	}
	inline poly operator *(poly a,poly b){
		int m=a.size()+b.size()-1,n=1;
		if(m<=128){
			poly c(m,0);
			for(int i=0;i<a.size();i++)
				for(int j=0;j<b.size();j++) inc(c[i+j],mul(a[i],b[j]));
			return c;	
		}
		while(n<m) n<<=1;
		init(n);
		a.resize(n);ntt(a,n,1);
		b.resize(n);ntt(b,n,1);
		for(int i=0;i<n;i++) Mul(a[i],b[i]);
		ntt(a,n,-1);a.resize(m);
		return a;
	}
}
using namespace Ntt;
int inv2;
inline poly Inv(poly a,int n){
	poly c,b(1,ksm(a[0],mod-2));
	for(int lim=4;lim<(n<<2);lim<<=1){
		init(lim);
		c=a;c.resize(lim>>1);
		c.resize(lim);ntt(c,lim,1);
		b.resize(lim);ntt(b,lim,1);
		for(int i=0;i<lim;i++) Mul(b[i],dec(2,mul(b[i],c[i])));
		ntt(b,lim,-1);b.resize(lim>>1);
	}
	b.resize(n);return b;
}
inline poly operator / (poly a,poly b){
	int lim=1,n=a.size()-b.size()+1;
	reverse(a.begin(),a.end());
	reverse(b.begin(),b.end());
	while(lim<=n) lim<<=1;
	b=Inv(b,lim);b.resize(n);
	a=a*b;a.resize(n);
	reverse(a.begin(),a.end());
	return a;
}
inline poly operator % (poly a,poly b){
	poly c=a-(a/b)*b;
	c.resize(b.size()-1);
	return c;
}
poly a,b,ans;
int main(){
	init_w();int n=read()+1,m=read()+1;a.resize(n);b.resize(m);
	for(int i=0;i<n;i++) a[i]=read();
	for(int i=0;i<m;i++) b[i]=read();
	ans=a/b;
	for(int i=0;i<n-m+1;i++) cout<<ans[i]<<" ";
	cout<<"\n";
	ans=a%b;
	for(int i=0;i<m-1;i++) cout<<ans[i]<<" ";
	return 0;
}

exp

#include<bits/stdc++.h>
#define poly vector<int>
#define ll long long
#define mod 998244353
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;
}
inline int add(int x,int y){x+=y;if(x>=mod) x-=mod;return x;}
inline int dec(int x,int y){x-=y;if(x<0) x+=mod;return x;}
inline int mul(int x,int y){return 1ll*x*y%mod;}
inline void Mul(int &x,int y){x=1ll*x*y%mod;}
inline void inc(int &x,int y){x+=y;if(x>=mod) x-=mod;}
inline int ksm(int a,int b){int res=1;for(;b;b>>=1,Mul(a,a)) if(b&1) Mul(res,a);return res;}
namespace Ntt{
	const int N=1e6+5;
	int *w[22],rev[N<<2];
	inline void init(int n){for(int i=0;i<n;i++) rev[i]=(rev[i>>1]>>1)|((i&1)*(n>>1));}
	inline void init_w(){
		for(int i=1;i<=21;i++) w[i]=new int[1<<(i-1)];
		int wn=ksm(3,(mod-1)/(1<<21));
		w[21][0]=1;
		for(int i=1;i<(1<<(20));i++) w[21][i]=mul(w[21][i-1],wn);
		for(int i=20;i;i--)
			for(int j=0;j<(1<<(i-1));j++) w[i][j]=w[i+1][j<<1];
	}
	inline void ntt(poly &f,int n,int kd){
		for(int i=0;i<n;i++) if(i>rev[i]) swap(f[i],f[rev[i]]);
		for(int mid=1,l=1;mid<n;mid<<=1,l++){
			for(int i=0;i<n;i+=(mid<<1)){
				for(int j=0,a0,a1;j<mid;j++){
					a0=f[i+j],a1=mul(f[i+j+mid],w[l][j]);
					f[i+j]=add(a0,a1);f[i+j+mid]=dec(a0,a1);
				}
			}
		}
		if(kd==-1 && (reverse(f.begin()+1,f.begin()+n),1))
			for(int inv=ksm(n,mod-2),i=0;i<n;i++) Mul(f[i],inv);
	}	
	inline poly operator -(poly a,poly b){
		poly c;int lim=max(a.size(),b.size());c.resize(lim);
		for(int i=0;i<lim;i++)c[i]=dec(a[i],b[i]);return c;
	}
	inline poly operator *(poly a,poly b){
		int m=a.size()+b.size()-1,n=1;
		if(m<=128){
			poly c(m,0);
			for(int i=0;i<a.size();i++)
				for(int j=0;j<b.size();j++) inc(c[i+j],mul(a[i],b[j]));
			return c;	
		}
		while(n<m) n<<=1;
		init(n);
		a.resize(n);ntt(a,n,1);
		b.resize(n);ntt(b,n,1);
		for(int i=0;i<n;i++) Mul(a[i],b[i]);
		ntt(a,n,-1);a.resize(m);
		return a;
	}
}
using namespace Ntt;
inline poly Inv(poly a,int n){
	poly c,b(1,ksm(a[0],mod-2));
	for(int lim=4;lim<(n<<2);lim<<=1){
		init(lim);
		c=a;c.resize(lim>>1);
		c.resize(lim);ntt(c,lim,1);
		b.resize(lim);ntt(b,lim,1);
		for(int i=0;i<lim;i++) Mul(b[i],dec(2,mul(b[i],c[i])));
		ntt(b,lim,-1);b.resize(lim>>1);
	}
	b.resize(n);return b;
}

inline poly deriv(poly a){
	for(int i=0;i<a.size()-1;i++) a[i]=mul(a[i+1],i+1);
	a.pop_back();return a;
}

ll inv[N];
inline poly integ(poly a){
	a.push_back(0);
	for(int i=a.size()-1;i;i--) a[i]=mul(a[i-1],inv[i]);
	a[0]=0;
	return a;
}
inline poly ln(poly a,int n){
	a=integ(deriv(a)*Inv(a,n)),a.resize(n);
	return a;
}

inline poly exp(poly a,int n){
	poly b(1,1),c;a.resize(n<<1);
	for(int lim=2;lim<(n<<1);lim<<=1){
		c=ln(b,lim);
		for(int i=0;i<lim;i++) c[i]=dec(a[i],c[i]);
		inc(c[0],1);b=b*c;
		b.resize(lim);
	}
	b.resize(n);return b;
}

poly a,ans;
int main(){
	inv[1]=1;
	for(int i=2;i<N;i++) inv[i]=((mod-mod/i)*inv[mod%i])%mod;
	init_w();int n=read();a.resize(n);
	for(int i=0;i<n;i++) a[i]=read();
	ans=exp(a,n);
	for(int i=0;i<n;i++) cout<<ans[i]<<" ";
	return 0;
}

inv求逆

#include<bits/stdc++.h>
#define poly vector<int>
#define ll long long
#define mod 998244353
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;
}
inline int add(int x,int y){x+=y;if(x>=mod) x-=mod;return x;}
inline int dec(int x,int y){x-=y;if(x<0) x+=mod;return x;}
inline int mul(int x,int y){return 1ll*x*y%mod;}
inline void Mul(int &x,int y){x=1ll*x*y%mod;}
inline void inc(int &x,int y){x+=y;if(x>=mod) x-=mod;}
inline int ksm(int a,int b){int res=1;for(;b;b>>=1,Mul(a,a)) if(b&1) Mul(res,a);return res;}
namespace Ntt{
	const int N=1e6+5;
	int *w[22],rev[N<<2];
	inline void init(int n){for(int i=0;i<n;i++) rev[i]=(rev[i>>1]>>1)|((i&1)*(n>>1));}
	inline void init_w(){
		for(int i=1;i<=21;i++) w[i]=new int[1<<(i-1)];
		int wn=ksm(3,(mod-1)/(1<<21));
		w[21][0]=1;
		for(int i=1;i<(1<<(20));i++) w[21][i]=mul(w[21][i-1],wn);
		for(int i=20;i;i--)
			for(int j=0;j<(1<<(i-1));j++) w[i][j]=w[i+1][j<<1];
	}
	inline void ntt(poly &f,int n,int kd){
		for(int i=0;i<n;i++) if(i>rev[i]) swap(f[i],f[rev[i]]);
		for(int mid=1,l=1;mid<n;mid<<=1,l++){
			for(int i=0;i<n;i+=(mid<<1)){
				for(int j=0,a0,a1;j<mid;j++){
					a0=f[i+j],a1=mul(f[i+j+mid],w[l][j]);
					f[i+j]=add(a0,a1);f[i+j+mid]=dec(a0,a1);
				}
			}
		}
		if(kd==-1 && (reverse(f.begin()+1,f.begin()+n),1))
			for(int inv=ksm(n,mod-2),i=0;i<n;i++) Mul(f[i],inv);
	}
	inline poly operator *(poly a,poly b){
		int m=a.size()+b.size()-1,n=1;
		if(m<=128){
			poly c(m,0);
			for(int i=0;i<a.size();i++)
				for(int j=0;j<b.size();j++) inc(c[i+j],mul(a[i],b[j]));
			return c;	
		}
		while(n<m) n<<=1;
		init(n);
		a.resize(n);ntt(a,n,1);
		b.resize(n);ntt(b,n,1);
		for(int i=0;i<n;i++) Mul(a[i],b[i]);
		ntt(a,n,-1);a.resize(m);
		return a;
	}
}
using namespace Ntt;
int inv2;
inline poly Inv(poly a,int n){
	poly c,b(1,ksm(a[0],mod-2));
	for(int lim=4;lim<(n<<2);lim<<=1){
		init(lim);
		c=a;c.resize(lim>>1);
		c.resize(lim);ntt(c,lim,1);
		b.resize(lim);ntt(b,lim,1);
		for(int i=0;i<lim;i++) Mul(b[i],dec(2,mul(b[i],c[i])));
		ntt(b,lim,-1);b.resize(lim>>1);
	}
	b.resize(n);return b;
}
inline poly sqr(poly a,int n){
	poly b(1,1),c,d;
	for(int lim=4;lim<(n<<2);lim<<=1){
		c=a;c.resize(lim>>1);
		init(lim);d=Inv(b,lim>>1);
		c.resize(lim);ntt(c,lim,1);
		d.resize(lim);ntt(d,lim,1);
		for(int i=0;i<lim;i++) Mul(c[i],d[i]);
		ntt(c,lim,-1);b.resize(lim>>1);
		for(int i=0;i<(lim>>1);i++) b[i]=mul(inv2,add(b[i],c[i]));
	}
	b.resize(n);return b;
}
poly a,ans;
int main(){
	inv2=ksm(2,mod-2);init_w();int n=read();a.resize(n);
	for(int i=0;i<n;i++) a[i]=read();
	ans=sqr(a,n);
	for(int i=0;i<n;i++) cout<<ans[i]<<" ";
	return 0;
}

ksm快速幂

#include<bits/stdc++.h>
#define poly vector<int>
#define ll long long
#define mod 998244353
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;
}
inline int add(int x,int y){x+=y;if(x>=mod) x-=mod;return x;}
inline int dec(int x,int y){x-=y;if(x<0) x+=mod;return x;}
inline int mul(int x,int y){return 1ll*x*y%mod;}
inline void Mul(int &x,int y){x=1ll*x*y%mod;}
inline void inc(int &x,int y){x+=y;if(x>=mod) x-=mod;}
inline int ksm(int a,int b){int res=1;for(;b;b>>=1,Mul(a,a)) if(b&1) Mul(res,a);return res;}

inline int ad(int x,int y){x+=y;if(x>=(mod-1)) x-=(mod-1);return x;}
inline int mu(int x,int y){return 1ll*x*y%(mod-1);}

namespace Ntt{
	const int N=1e6+5;
	int *w[22],rev[N<<2];
	inline void init(int n){for(int i=0;i<n;i++) rev[i]=(rev[i>>1]>>1)|((i&1)*(n>>1));}
	inline void init_w(){
		for(int i=1;i<=21;i++) w[i]=new int[1<<(i-1)];
		int wn=ksm(3,(mod-1)/(1<<21));
		w[21][0]=1;
		for(int i=1;i<(1<<(20));i++) w[21][i]=mul(w[21][i-1],wn);
		for(int i=20;i;i--)
			for(int j=0;j<(1<<(i-1));j++) w[i][j]=w[i+1][j<<1];
	}
	inline void ntt(poly &f,int n,int kd){
		for(int i=0;i<n;i++) if(i>rev[i]) swap(f[i],f[rev[i]]);
		for(int mid=1,l=1;mid<n;mid<<=1,l++){
			for(int i=0;i<n;i+=(mid<<1)){
				for(int j=0,a0,a1;j<mid;j++){
					a0=f[i+j],a1=mul(f[i+j+mid],w[l][j]);
					f[i+j]=add(a0,a1);f[i+j+mid]=dec(a0,a1);
				}
			}
		}
		if(kd==-1 && (reverse(f.begin()+1,f.begin()+n),1))
			for(int inv=ksm(n,mod-2),i=0;i<n;i++) Mul(f[i],inv);
	}	
	inline poly operator -(poly a,poly b){
		poly c;int lim=max(a.size(),b.size());c.resize(lim);
		for(int i=0;i<lim;i++)c[i]=dec(a[i],b[i]);return c;
	}
	inline poly operator *(poly a,int b){
		poly c;int lim=a.size();c.resize(lim);
		for(int i=0;i<lim;i++) c[i]=mul(a[i],b);return c;
	}
	inline poly operator *(poly a,poly b){
		int m=a.size()+b.size()-1,n=1;
		if(m<=128){
			poly c(m,0);
			for(int i=0;i<a.size();i++)
				for(int j=0;j<b.size();j++) inc(c[i+j],mul(a[i],b[j]));
			return c;	
		}
		while(n<m) n<<=1;
		init(n);
		a.resize(n);ntt(a,n,1);
		b.resize(n);ntt(b,n,1);
		for(int i=0;i<n;i++) Mul(a[i],b[i]);
		ntt(a,n,-1);a.resize(m);
		return a;
	}
}
using namespace Ntt;
inline poly Inv(poly a,int n){
	poly c,b(1,ksm(a[0],mod-2));
	for(int lim=4;lim<(n<<2);lim<<=1){
		init(lim);
		c=a;c.resize(lim>>1);
		c.resize(lim);ntt(c,lim,1);
		b.resize(lim);ntt(b,lim,1);
		for(int i=0;i<lim;i++) Mul(b[i],dec(2,mul(b[i],c[i])));
		ntt(b,lim,-1);b.resize(lim>>1);
	}
	b.resize(n);return b;
}

inline poly deriv(poly a){
	for(int i=0;i<a.size()-1;i++) a[i]=mul(a[i+1],i+1);
	a.pop_back();return a;
}

ll inv[N];
inline poly integ(poly a){
	a.push_back(0);
	for(int i=a.size()-1;i;i--) a[i]=mul(a[i-1],inv[i]);
	a[0]=0;
	return a;
}
inline poly ln(poly a,int n){
	a=integ(deriv(a)*Inv(a,n)),a.resize(n);
	return a;
}

inline poly exp(poly a,int n){
	poly b(1,1),c;a.resize(n<<1);
	for(int lim=2;lim<(n<<1);lim<<=1){
		c=ln(b,lim);
		for(int i=0;i<lim;i++) c[i]=dec(a[i],c[i]);
		inc(c[0],1);b=b*c;
		b.resize(lim);
	}
	b.resize(n);return b;
}

inline poly ksm(poly a,int n,int k){
	a=exp(ln(a,n)*k,n),a.resize(n);
	return a;
}

inline int readk(){
	char c;
	while(!isdigit(c=getchar()));int num=c^48;
	while(isdigit(c=getchar()))num=add(mul(num,10),c^48);
	return num;
}
poly a,ans;
int main(){
	inv[0]=inv[1]=1;
	for(int i=2;i<N;i++) inv[i]=mul((mod-mod/i),inv[mod%i]);
	init_w();int n=read(),k=readk();a.resize(n);
	for(int i=0;i<n;i++) a[i]=read();
	ans=ksm(a,n,k);
	for(int i=0;i<n;i++) cout<<ans[i]<<" ";
	return 0;
}

ln

#include<bits/stdc++.h>
#define poly vector<int>
#define ll long long
#define mod 998244353
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;
}
inline int add(int x,int y){x+=y;if(x>=mod) x-=mod;return x;}
inline int dec(int x,int y){x-=y;if(x<0) x+=mod;return x;}
inline int mul(int x,int y){return 1ll*x*y%mod;}
inline void Mul(int &x,int y){x=1ll*x*y%mod;}
inline void inc(int &x,int y){x+=y;if(x>=mod) x-=mod;}
inline int ksm(int a,int b){int res=1;for(;b;b>>=1,Mul(a,a)) if(b&1) Mul(res,a);return res;}
namespace Ntt{
	const int N=1e6+5;
	int *w[22],rev[N<<2];
	inline void init(int n){for(int i=0;i<n;i++) rev[i]=(rev[i>>1]>>1)|((i&1)*(n>>1));}
	inline void init_w(){
		for(int i=1;i<=21;i++) w[i]=new int[1<<(i-1)];
		int wn=ksm(3,(mod-1)/(1<<21));
		w[21][0]=1;
		for(int i=1;i<(1<<(20));i++) w[21][i]=mul(w[21][i-1],wn);
		for(int i=20;i;i--)
			for(int j=0;j<(1<<(i-1));j++) w[i][j]=w[i+1][j<<1];
	}
	inline void ntt(poly &f,int n,int kd){
		for(int i=0;i<n;i++) if(i>rev[i]) swap(f[i],f[rev[i]]);
		for(int mid=1,l=1;mid<n;mid<<=1,l++){
			for(int i=0;i<n;i+=(mid<<1)){
				for(int j=0,a0,a1;j<mid;j++){
					a0=f[i+j],a1=mul(f[i+j+mid],w[l][j]);
					f[i+j]=add(a0,a1);f[i+j+mid]=dec(a0,a1);
				}
			}
		}
		if(kd==-1 && (reverse(f.begin()+1,f.begin()+n),1))
			for(int inv=ksm(n,mod-2),i=0;i<n;i++) Mul(f[i],inv);
	}	
	inline poly operator -(poly a,poly b){
		poly c;int lim=max(a.size(),b.size());c.resize(lim);
		for(int i=0;i<lim;i++)c[i]=dec(a[i],b[i]);return c;
	}
	inline poly operator *(poly a,poly b){
		int m=a.size()+b.size()-1,n=1;
		if(m<=128){
			poly c(m,0);
			for(int i=0;i<a.size();i++)
				for(int j=0;j<b.size();j++) inc(c[i+j],mul(a[i],b[j]));
			return c;	
		}
		while(n<m) n<<=1;
		init(n);
		a.resize(n);ntt(a,n,1);
		b.resize(n);ntt(b,n,1);
		for(int i=0;i<n;i++) Mul(a[i],b[i]);
		ntt(a,n,-1);a.resize(m);
		return a;
	}
}
using namespace Ntt;
inline poly Inv(poly a,int n){
	poly c,b(1,ksm(a[0],mod-2));
	for(int lim=4;lim<(n<<2);lim<<=1){
		init(lim);
		c=a;c.resize(lim>>1);
		c.resize(lim);ntt(c,lim,1);
		b.resize(lim);ntt(b,lim,1);
		for(int i=0;i<lim;i++) Mul(b[i],dec(2,mul(b[i],c[i])));
		ntt(b,lim,-1);b.resize(lim>>1);
	}
	b.resize(n);return b;
}

inline poly deriv(poly a){
	for(int i=0;i<a.size()-1;i++) a[i]=mul(a[i+1],i+1);
	a.pop_back();return a;
}

ll inv[N];
inline poly integ(poly a){
	a.push_back(0);
	for(int i=a.size()-1;i;i--) a[i]=mul(a[i-1],inv[i]);
	a[0]=0;
	return a;
}
inline poly ln(poly a,int n){
	a=integ(deriv(a)*Inv(a,n)),a.resize(n);
	return a;
}

poly a,ans;
int main(){
	inv[1]=1;
	for(int i=2;i<N;i++) inv[i]=((mod-mod/i)*inv[mod%i])%mod;
	init_w();int n=read();a.resize(n);
	for(int i=0;i<n;i++) a[i]=read();
	ans=ln(a,n);
	for(int i=0;i<n;i++) cout<<ans[i]<<" ";
	return 0;
}

sqr开方

#include<bits/stdc++.h>
#define poly vector<int>
#define ll long long
#define mod 998244353
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;
}
inline int add(int x,int y){x+=y;if(x>=mod) x-=mod;return x;}
inline int dec(int x,int y){x-=y;if(x<0) x+=mod;return x;}
inline int mul(int x,int y){return 1ll*x*y%mod;}
inline void Mul(int &x,int y){x=1ll*x*y%mod;}
inline void inc(int &x,int y){x+=y;if(x>=mod) x-=mod;}
inline int ksm(int a,int b){int res=1;for(;b;b>>=1,Mul(a,a)) if(b&1) Mul(res,a);return res;}
namespace Ntt{
	const int N=1e6+5;
	int *w[22],rev[N<<2];
	inline void init(int n){for(int i=0;i<n;i++) rev[i]=(rev[i>>1]>>1)|((i&1)*(n>>1));}
	inline void init_w(){
		for(int i=1;i<=21;i++) w[i]=new int[1<<(i-1)];
		int wn=ksm(3,(mod-1)/(1<<21));
		w[21][0]=1;
		for(int i=1;i<(1<<(20));i++) w[21][i]=mul(w[21][i-1],wn);
		for(int i=20;i;i--)
			for(int j=0;j<(1<<(i-1));j++) w[i][j]=w[i+1][j<<1];
	}
	inline void ntt(poly &f,int n,int kd){
		for(int i=0;i<n;i++) if(i>rev[i]) swap(f[i],f[rev[i]]);
		for(int mid=1,l=1;mid<n;mid<<=1,l++){
			for(int i=0;i<n;i+=(mid<<1)){
				for(int j=0,a0,a1;j<mid;j++){
					a0=f[i+j],a1=mul(f[i+j+mid],w[l][j]);
					f[i+j]=add(a0,a1);f[i+j+mid]=dec(a0,a1);
				}
			}
		}
		if(kd==-1 && (reverse(f.begin()+1,f.begin()+n),1))
			for(int inv=ksm(n,mod-2),i=0;i<n;i++) Mul(f[i],inv);
	}
	inline poly operator *(poly a,poly b){
		int m=a.size()+b.size()-1,n=1;
		if(m<=128){
			poly c(m,0);
			for(int i=0;i<a.size();i++)
				for(int j=0;j<b.size();j++) inc(c[i+j],mul(a[i],b[j]));
			return c;	
		}
		while(n<m) n<<=1;
		init(n);
		a.resize(n);ntt(a,n,1);
		b.resize(n);ntt(b,n,1);
		for(int i=0;i<n;i++) Mul(a[i],b[i]);
		ntt(a,n,-1);a.resize(m);
		return a;
	}
}
using namespace Ntt;
int inv2;
inline poly Inv(poly a,int n){
	poly c,b(1,ksm(a[0],mod-2));
	for(int lim=4;lim<(n<<2);lim<<=1){
		init(lim);
		c=a;c.resize(lim>>1);
		c.resize(lim);ntt(c,lim,1);
		b.resize(lim);ntt(b,lim,1);
		for(int i=0;i<lim;i++) Mul(b[i],dec(2,mul(b[i],c[i])));
		ntt(b,lim,-1);b.resize(lim>>1);
	}
	b.resize(n);return b;
}
inline poly sqr(poly a,int n){
	poly b(1,1),c,d;
	for(int lim=4;lim<(n<<2);lim<<=1){
		c=a;c.resize(lim>>1);
		init(lim);d=Inv(b,lim>>1);
		c.resize(lim);ntt(c,lim,1);
		d.resize(lim);ntt(d,lim,1);
		for(int i=0;i<lim;i++) Mul(c[i],d[i]);
		ntt(c,lim,-1);b.resize(lim>>1);
		for(int i=0;i<(lim>>1);i++) b[i]=mul(inv2,add(b[i],c[i]));
	}
	b.resize(n);return b;
}
poly a,ans;
int main(){
	inv2=ksm(2,mod-2);init_w();int n=read();a.resize(n);
	for(int i=0;i<n;i++) a[i]=read();
	ans=sqr(a,n);
	for(int i=0;i<n;i++) cout<<ans[i]<<" ";
	return 0;
}

模板以洛谷多项式模板为准

romiqi_new
关注
  • 1
    点赞
  • 踩
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
专栏目录
ACM多项式算法入门
CCCCTong的博客
03-30 768
在摸索了很久很久，还是没找到多项式的正确入门方式，所以尝试着自己写一下 多项式乘法 对于多项式A(x),B(x)A(x),B(x)A(x),B(x) A(x)=anxn+an−1xn−1+an−2xn−2+an−3xn−3+...+a1x+a0A(x)=a_nx^n + a_{n-1}x ^ {n-1} + a_{n-2}x ^ {n-2} + a_{n-3}x ^ {n-3} + ... + ...
matlab数学建模算法模板
02-07
"matlab数学建模算法模板"是一个集合了基础MATLAB算法实现的资源，对于初学者和进阶者来说，都是一个极具价值的学习资料。下面将详细介绍这些模板中的关键知识点。 首先，MATLAB作为一门编程语言，其语法简洁明了，...
P5282 【模板】快速阶乘算法（多项式运算+拉格朗日插值+倍增）
weixin_33978016的博客
04-06 501
题面 传送门 前置芝士 优化后的\(MTT\)（四次\(FFT\)） 题解 这里有多点求值的做法然而被\(shadowice\)巨巨吊起来打了一顿，所以来学一下倍增 成功同时拿到本题最优解和最劣解…… \(Min_{25}\)牛逼！（据说这是原文然而我看不懂就是了） 真的快的不要不要的…… 和多点求值一样，我们还是设\(s=\sqrt{n}\)，并设多项式 \[g_s(x)=\sum_{i=1}^...
多项式模板
weixin_30246221的博客
02-25 76
模板1 int版 模板2 拆系数FFT 模板3 LL版 模板4 多点求值 int版 模板1 #include <bits/stdc++.h> using namespace std; typedef long long LL; const int MOD = 998244353; inline int Add(int a, int b) { return ...
多项式系列模板
Never give in.
04-23 711
FFT struct Z { double x,y; Z(double _x=0,double _y=0):x(_x),y(_y){}; }; Z operator +(Z x,Z y) {return Z(x.x+y.x,x.y+y.y);} Z operator -(Z x,Z y) {return Z(x.x-y.x,x.y-y.y);} Z operator *(Z ...
多项式算法合集
Stargazer的博客
08-03 884
文章目录多项式加减点乘点除多项式乘法FFT：前置算法代码实现NTT：代码实现多项式求逆：求解过程代码实现多项式开方：求解过程代码实现多项式除法和取模：求解过程代码实现多项式求导与积分代码实现多项式Ln求解过程代码实现多项式Exp前置知识泰勒展开牛顿迭代求解过程代码实现多项式多点求值求解过程代码实现多项式快速插值下降幂多项式乘法其他技巧多项式快速幂分治NTTBluestein混合基NTTMTT模板合...
多项式算法7：多项式除法
ha_ing的博客
07-07 696
多项式算法7：多项式除法
多项式算法6：分治 FFT
ha_ing的博客
07-06 725
多项式算法6：分治 FFT
多项式算法5：多项式求逆
ha_ing的博客
06-03 2513
多项式算法5：多项式求逆
多项式算法2：NTT（快速数论变换）
ha_ing的博客
05-11 2340
多项式算法2：NTT（快速数论变换）
算法文档无代码多项式乘法算法文档无代码多项式乘法
04-14
算法文档无代码多项式乘法算法文档无代码多项式乘法提取方式是百度网盘分享地址
Dinic算法标准模板
10-15
Dinic算法的时间复杂度为O(mn2)，其中m为弧的数目，是多项式算法。邻接表表示图，空间复杂度为O(n+m)。Dinic算法的思想是为了减少增广次数，建立一个辅助网络L，L与原网络G具有相同的节点数，但边上的容量有所不同，...
稀疏多项式
12-17
稀疏多项式是一种在计算机科学和数学中处理大型多项式表达式时常用的数据结构，尤其在算法和计算复杂性理论中扮演着重要角色。在实际应用中，如果一个多项式的系数大多数为零，那么使用传统的数组表示法会浪费大量...
c++写的多项式相乘算法
08-07
本篇将详细介绍C++中实现多项式相乘的算法及其相关的数据结构与编程技巧。 首先，我们需要定义一个数据结构来表示多项式。一种常见的方法是使用链表或数组来存储多项式的系数和指数。这里我们可以选择使用动态数组...
励志2024龙年筑梦前行共创未来PPT模板，适用于工作计划，工作总结
07-15
【作品名称】：励志2024龙年筑梦前行共创未来PPT模板，适用于工作计划，工作总结 【适用人群】：适用于希望学习不同技术领域的小白或进阶学习者。可作为毕设项目、课程设计、大作业、工程实训或初期项目立项。
内蒙古医科大学考研17个学院，57个专业课历年考试真题及答案汇总，备考资料题库笔记，专业课调剂信息查询.pdf
07-15
内蒙古医科大学考研17个学院，57个专业课历年考试真题及答案汇总，备考资料题库笔记，专业课调剂信息查询.pdf
齐齐哈尔医学院考研，57个专业课历年考试真题及答案汇总整理，备考资料题库视频.pdf
最新发布
07-15
齐齐哈尔医学院考研，57个专业课历年考试真题及答案汇总整理，备考资料题库视频.pdf
长三角城市及各区县数据集.txt
07-15
因文件较多，数据存放网盘，txt文件内包含下载链接及提取码，永久有效。失效会第一时间进行补充。样例数据及详细介绍参见文章：https://blog.csdn.net/m0_65541699/article/details/140428710
原子轨道搜索算法模板matlab
05-15
以下是一个简单的MATLAB代码模板，用于搜索原子轨道。 ``` clear all; clc; % 定义系统参数 Z = 1; % 原子核电荷数 ...注意：此代码仅提供一个基本的...同时，还需要添加LaguerreL函数的实现，用于计算Laguerre多项式。

“相关推荐”对你有帮助么？

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

个

红包个数最小为10个

元

红包金额最低5元

当前余额3.43元 前往充值 >
需支付：10.00元
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付元
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明：

1.余额是钱包充值的虚拟货币，按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载，可以购买VIP、付费专栏及课程。

余额充值