多项式求逆

设已知A(x)B(x)=1(mod x^n),考虑A(x)C(x)=1(mod x^2n)

显然B(x)-C(x)=0(mod x^n)

平方得B(x)^2-2B(x)C(x)+C(x)^2=0(mod x^2n)

同乘A(x)得A(x)B(x)^2-2B(x)+C(x)=0(mod x^2n)

即有C(x)=2B(x)-A(x)B(x)^2,fft即可

设A(x)常数项为t,则A(x)*1/t=1(mod x^1)/

不断倍增即可解决

注意fft又是循环卷积,实现的时候必须做适当的清零

然后每个式子在不全相同的模意义下成立,清零的区间是哪段也要注意一下

代码:

#include<cstdio>
namespace poly{
	#define mo 998244353
	struct AwD{int x;};
	AwD operator+(AwD a,AwD b){return (AwD){(a.x+b.x)%mo};}
	AwD operator-(AwD a,AwD b){return (AwD){(a.x-b.x+mo)%mo};}
	AwD operator*(AwD a,AwD b){return (AwD){(int)(1LL*a.x*b.x%mo)};}
	AwD operator^(AwD a,int b){if(b<0) b+=mo-1;if(!b) return (AwD){1};AwD temp=a^(b>>1);temp=temp*temp;if(b&1) temp=temp*a;return temp;}
	AwD operator/(AwD a,AwD b){return a*(b^-1);}
	const AwD root=(AwD){3};
	const int om=mo-1;
	void ntt(AwD*a,int n,int d){
		int i,j,k;
		AwD w,t,u,v;
		for(i=(n>>1),j=1;j<n;j++){
			if(i<j) t=a[i],a[i]=a[j],a[j]=t;
			for(k=(n>>1);i&k;i^=k,k>>=1);i^=k;
		}
		for(k=2;k<=n;k<<=1){
			w=root^((mo-1)/k*d);
			for(i=0;i<n;i+=k){
				t=(AwD){1};
				for(j=i;j<i+(k>>1);j++){
					u=a[j];v=t*a[j+(k>>1)];
					a[j]=u+v;a[j+(k>>1)]=u-v;t=t*w;
				}
			}
		}
	}
	AwD a[1<<20],b[1<<20];
	void print(AwD*a,int l){
		for(int i=0;i<l;i++) printf("%d ",a[i].x);
		printf("\n");
	}
	void plus(AwD*_a,AwD*_b,int l,AwD*c){
		for(int i=0;i<l;i++) a[i]=_a[i],b[i]=_b[i];
		for(int i=0;i<l;i++) c[i]=a[i]+b[i];
	}
	void subt(AwD*_a,AwD*_b,int l,AwD*c){
		for(int i=0;i<l;i++) a[i]=_a[i],b[i]=_b[i];
		for(int i=0;i<l;i++) c[i]=a[i]-b[i];
	}
	void mult(AwD*_a,AwD b,int l,AwD*c){
		for(int i=0;i<l;i++) a[i]=_a[i];
		for(int i=0;i<l;i++) c[i]=a[i]*b;
	}
	void mult(AwD*_a,AwD*_b,int l,AwD*c){
		for(int i=0;i<l;i++) a[i]=_a[i],b[i]=_b[i];
		ntt(a,l,1);ntt(b,l,1);
		for(int i=0;i<l;i++) c[i]=a[i]*b[i];
		ntt(c,l,-1);
		for(int i=0;i<l;i++) c[i]=c[i]/(AwD){l};
	}
	AwD a1[1<<20],aa[1<<20],tmp[1<<20];
	void inv(AwD*_a,int l,AwD*b){	
		for(int i=0;i<l;i++) a1[i]=_a[i];
		for(int i=0;i<l;i++) b[i]=i?(AwD){0}:a1[i]^-1;
		for(int l0=2;l0<=l;l0<<=1){
			mult(b,(AwD){2},l0>>1,tmp);
			mult(b,b,l0,b);
			for(int i=0;i<(l0<<1);i++) aa[i]=i<l0?a1[i]:(AwD){0};
			mult(aa,b,l0<<1,b);
			for(int i=l0;i<(l0<<1);i++) b[i]=(AwD){0};
			subt(tmp,b,l0,b);
		}
	}
}
int n,l;
poly::AwD a[1<<20];
int main(){
	scanf("%d",&n);n++;
	for(int i=0;i<n;i++) scanf("%d",&a[i].x);
	l=1;while(l<n) l<<=1;for(int i=n;i<l;i++) a[i].x=0;
	poly::inv(a,l,a);
	for(int i=0;i<n;i++) printf("%d ",a[i].x);
}

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值