推导过程看楼上的博客就可以了。
注意事项
在对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;
}