分治NTT

最新推荐文章于 2024-01-26 02:49:59 发布

gigo_64

最新推荐文章于 2024-01-26 02:49:59 发布

阅读量974

点赞数

分类专栏： ----------数论------------- 多项式

本文链接：https://blog.csdn.net/qq_42835815/article/details/102508022

版权

----------数论------------- 同时被 2 个专栏收录

36 篇文章 0 订阅

订阅专栏

多项式

12 篇文章 0 订阅

订阅专栏

以前是背板今天是理解。

前置思想CDQ。

当一个多项式前面对后面有贡献，递归分治处理。

跟CDQ一样，先左边，然后左边更新右边，然后右边。

考虑下面这个图

l ----- mid ----- r

xxxxxxxxxxxxxxxxx系数数组

xxxxxx xxxxxxx答案数组

假设我们已经求得答案数组左边，用答案数组前mid-l+1和系数数组的前r-l+1项卷积即可得到后面。系数依次对应加入答案数组。

记得初始化临时数组。

#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;
}
int a[1000003],b[1000003],c[1000003],d[1000003];
int limit,l,r[1000003],n;
void getl(int len){
	limit=1,l=0;
	while(limit<len)limit<<=1,l++;
	for(int i=0;i<limit;i++){
		r[i]=(r[i>>1]>>1)|((i&1)<<(l-1));
	}
}const int mod=998244353;
int ksm(int a,int b){
	int sum=1;
	while(b){
		if(b&1)sum=sum*a%mod;a=a*a%mod;b>>=1;
	}return sum;
}
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,(mod-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%mod){
				int x=a[i+j],y=a[i+j+mid]*omega%mod;
				a[i+j]=(x+y)%mod;a[i+j+mid]=(x-y+mod)%mod;
			}
		}
	}if(inv==1)return;
	int gu=ksm(len,mod-2);reverse(a+1,a+len);
	for(int i=0;i<len;i++)a[i]=a[i]*gu%mod;
}
void CDQ(int l,int r){
	if(l==r){
		return;
	}
	int mid=(l+r)>>1;CDQ(l,mid);
	getl((r-l+1)<<1);
	for(int i=0;i<limit;i++)c[i]=d[i]=0;
	for(int i=l;i<=r;i++)c[i-l]=a[i-l];
	for(int i=l;i<=mid;i++)d[i-l]=b[i];
	NTT(c,limit,1);NTT(d,limit,1);
	for(int i=0;i<limit;i++)c[i]=c[i]*d[i]%mod;
	NTT(c,limit,-1);
	for(int i=mid+1;i<=r;i++)b[i]=(b[i]+c[i-l])%mod;
	CDQ(mid+1,r);
}

signed main(){
	n=in;
	for(int i=1;i<n;i++)a[i]=in;
	b[0]=1;CDQ(0,n-1);
	for(int i=0;i<n;i++)printf("%lld ",b[i]);
	return 0;
}