多项式求逆

我低头,你踮脚,刚刚好

推导过程看楼上的博客就可以了。

注意事项

在对NTT进行IDFT的时候,本来wn应该要变成自己%len意义下的逆元。但根据zxy说的一位大佬的证明,和zxyoi的感性理解我们发现,只需要把原来的所有含x项翻转就可以了。

记得在做limit的时候要做到deg*2,防止溢出。并且开c数组来替代a数组防止出现错误。

一个跟博客几乎一样的代码

#include<bits/stdc++.h>
using namespace std;
#define in read()
#define int long long
int in{
	int cnt=0,f=1;char ch=0;
	while(!isdigit(ch)){
		ch=getchar();if(ch=='-')f=-1;
	}
	while(isdigit(ch)){
		cnt=cnt*10+ch-48;
		ch=getchar();
	}return cnt*f;
}
const int p=998244353;
int ksm(int a,int b){
	int sum=1;
	while(b){
		if(b&1)sum=sum*a%p;
		a=a*a%p;b>>=1;
	}return sum;
}
int a[2000003],b[2000003],c[2000003],r[2000003];
void NTT(int *A,int len,int inv){
	for(int i=0;i<len;i++){if(i<r[i])swap(A[i],A[r[i]]);}
	for(int mid=1;mid<len;mid<<=1){
		int wn=ksm(3,(p-1)/(mid<<1));
		for(int i=0;i<len;i+=mid*2){
			int omega=1;
			for(int j=0;j<mid;j++,omega=omega*wn%p){
				int x=A[i+j],y=A[i+j+mid]*omega%p;
				A[i+j]=x+y;A[i+j]%=p;A[i+j+mid]=x-y;A[i+j+mid]=(A[i+j+mid]%p+p)%p;
			}
		}
	}
	if(inv==-1){
		int gu=ksm(len,p-2);reverse(A+1,A+len);
		for(int i=0;i<len;i++)A[i]=A[i]*gu%p;
	}
}
void work(int deg,int *a,int *b){
	if(deg==1){b[0]=ksm(a[0],p-2);return;}
	work((deg+1)>>1,a,b);
	int limit=1,l=0;
	while(limit<(deg<<1))limit<<=1,l++;
	for(int i=1;i<limit;i++)r[i]=(r[i>>1]>>1)|((i&1)<<(l-1));
	for(int i=0;i<deg;i++)c[i]=a[i];for(int i=deg;i<limit;i++)c[i]=0;
	NTT(c,limit,1);NTT(b,limit,1);
	for(int i=0;i<limit;i++)b[i]=((2-c[i]*b[i]%p)+p)%p*b[i]%p;
	NTT(b,limit,-1);
	for(int i=deg;i<limit;i++)b[i]=0;
}
int n;
signed main(){
	n=in;
	for(int i=0;i<n;i++)a[i]=in;work(n,a,b);
	for(int i=0;i<n;i++)printf("%lld ",b[i]);
	return 0;
}

 

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值