ICode9

精准搜索请尝试: 精确搜索
首页 > 其他分享> 文章详细

多项式工业基础与全家桶

2022-05-02 11:35:22  阅读:144  来源: 互联网

标签:int 多项式 全家 1ll len ret Poly 工业 mod


多项式工业基础与全家桶

开坑待填,放个常数巨大的板子先

#define Maxn 200005
#define mod 998244353
inline int ksm(int x,int y=mod-2)
{
	int ret=1;
	for(;y;y>>=1,x=1ll*x*x%mod) if(y&1) ret=1ll*ret*x%mod;
	return ret;
}
const int G=3,invG=ksm(3,mod-2);
int tr[Maxn<<1];
struct Poly
{
	int len,f[Maxn<<1];
	Poly(int L=0) { len=L,memset(f,0,sizeof(f)); }
	inline void ntt(int n,int opt)
	{
		for(int i=0;i<n;i++) tr[i]=(tr[i>>1]>>1)|((i&1)?(n>>1):0);
		for(int i=0;i<n;i++) if(i<tr[i]) swap(f[i],f[tr[i]]);
		for(int p=2;p<=n;p<<=1)
		{
			int tG=ksm((opt==1)?G:invG,(mod-1)/p),L=p>>1;
			for(int i=0;i<n;i+=p)
			{
				int buf=1;
				for(int j=0;j<L;j++)
				{
					int tmp=1ll*buf*f[i+j+L]%mod;
					f[i+j+L]=f[i+j]-tmp;
					if(f[i+j+L]<0) f[i+j+L]+=mod;
					f[i+j]=f[i+j]+tmp;
					if(f[i+j]>=mod) f[i+j]-=mod;
					buf=1ll*buf*tG%mod;
				}
			}
		}
	}
	Poly friend operator + (Poly x,Poly y)
	{
		x.len=max(x.len,y.len);
		for(int i=0;i<=x.len;i++) x.f[i]+=y.f[i];
		for(int i=0;i<=x.len;i++) if(x.f[i]>=mod) x.f[i]-=mod;
		return x;
	}
	Poly friend operator - (Poly x,Poly y)
	{
		x.len=max(x.len,y.len);
		for(int i=0;i<=x.len;i++) x.f[i]-=y.f[i];
		for(int i=0;i<=x.len;i++) if(x.f[i]<0) x.f[i]+=mod;
		return x;
	}
	Poly friend operator * (Poly x,int y)
	{
		for(int i=0;i<=x.len;i++) x.f[i]=1ll*x.f[i]*y%mod;
		return x;
	}
	Poly friend operator * (Poly x,Poly y)
	{
		int L=1;
		for(;L<=(x.len+y.len);L<<=1);
		int invl=ksm(L);
		x.ntt(L,1),y.ntt(L,1);
		for(int i=0;i<L;i++) x.f[i]=1ll*x.f[i]*y.f[i]%mod;
		x.ntt(L,-1);
		for(int i=0;i<L;i++) x.f[i]=1ll*x.f[i]*invl%mod;
		x.len=x.len+y.len;
		return x;
	}
	Poly friend operator % (Poly x,int y)
	{
		for(int i=y+1;i<=x.len;i++) x.f[i]=0;
		x.len=y;
		return x;
	}
	Poly invf()
	{
		Poly ret(0);
		ret.f[0]=ksm(f[0]);
		int L=1; for(;L<len;L<<=1);
		for(int p=2;p<=L;p<<=1) ret=((ret*2)-(*this%(p-1))*ret*ret)%(p-1);
		return ret;
	}
	Poly friend operator / (Poly x,Poly y)
	{
		y=y.invf();
		int L=1;
		for(;L<=(x.len+y.len);L<<=1);
		int invl=ksm(L);
		x.ntt(L,1),y.ntt(L,1);
		for(int i=0;i<L;i++) x.f[i]=1ll*x.f[i]*y.f[i]%mod;
		x.ntt(L,-1);
		for(int i=0;i<L;i++) x.f[i]=1ll*x.f[i]*invl%mod;
		x.len=x.len+y.len;
		return x;
	}
};

标签:int,多项式,全家,1ll,len,ret,Poly,工业,mod
来源: https://www.cnblogs.com/EricQian/p/16215354.html

本站声明: 1. iCode9 技术分享网(下文简称本站)提供的所有内容,仅供技术学习、探讨和分享;
2. 关于本站的所有留言、评论、转载及引用,纯属内容发起人的个人观点,与本站观点和立场无关;
3. 关于本站的所有言论和文字,纯属内容发起人的个人观点,与本站观点和立场无关;
4. 本站文章均是网友提供,不完全保证技术分享内容的完整性、准确性、时效性、风险性和版权归属;如您发现该文章侵犯了您的权益,可联系我们第一时间进行删除;
5. 本站为非盈利性的个人网站,所有内容不会用来进行牟利,也不会利用任何形式的广告来间接获益,纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。

专注分享技术,共同学习,共同进步。侵权联系[81616952@qq.com]

Copyright (C)ICode9.com, All Rights Reserved.

ICode9版权所有