P4721-分治FFT,生成函数

P4721

题目描述

题目描述

题解

两种解法
1,分治FFT O ( n l o g 2 n ) O(nlog^2n) O(nlog2n)
考虑直接FFT,由于每次都要FFT一次,所以时间复杂度退化到 O ( n 2 l o g n ) O(n^2logn) O(n2logn)
考虑优化,类似于CDQ分治。
对于一个区间 [ l , r ] [l,r] [l,r]之间的答案,我们先处理出 [ l , m i d ] [l,mid] [l,mid]的答案,然后算出左区间对右区间的贡献,再递归处理 [ m i d + 1 , r ] [mid+1,r] [mid+1,r]
那么左区间对右区间的贡献即是:
f [ i ] = ∑ j = l m i d f [ j ] ∗ g [ i − j ] , i ∈ [ m i d + 1 , r ] f[i]=\sum\limits_{j=l}^{mid}f[j]*g[i-j],i\in[mid+1,r] f[i]=j=lmidf[j]g[ij],i[mid+1,r]
所以有 l o g n logn logn层,每层n个元素,FFT复杂度为 O ( n l o g n ) O(nlogn) O(nlogn),总复杂度即为 O ( n l o g 2 n ) O(nlog^2n) O(nlog2n)
2,生成函数
f n , g n f_n,g_n fn,gn的生成函数为 F ( x ) , G ( x ) ( g [ 0 ] = 0 ) F(x),G(x)(g[0]=0) F(x),G(x)(g[0]=0)
则有
F ( x ) = ∑ i = 0 ∞ f i x i , G ( x ) = ∑ i = 0 ∞ g i x i F ( x ) ∗ G ( x ) = ∑ i = 0 ∞ ∑ j = 0 ∞ f i g j x i + j = ∑ i = 0 ∞ ∑ j = 0 i f i g i − j x i = F ( x ) + f [ 0 ] F(x)=\sum\limits_{i=0}^{\infty}f_ix^i,G(x)=\sum\limits_{i=0}^{\infty}g_ix^i\\ F(x)*G(x)=\sum\limits_{i=0}^{\infty}\sum\limits_{j=0}^{\infty}f_ig_jx^{i+j}=\sum\limits_{i=0}^{\infty}\sum\limits_{j=0}^{i}f_ig_{i-j}x^i=F(x)+f[0]\\ F(x)=i=0fixi,G(x)=i=0gixiF(x)G(x)=i=0j=0figjxi+j=i=0j=0ifigijxi=F(x)+f[0]
由题干所给式子, F ( x ) ∗ G ( x ) F(x)*G(x) F(x)G(x)可以化成最后那部分,由于 g [ 0 ] = 0 g[0]=0 g[0]=0,所以常数项变成了 0 0 0,故还要加上 f [ 0 ] f[0] f[0]
F ( x ) = f [ 0 ] 1 − G ( x ) = 1 1 − G ( x ) F(x)=\frac{f[0]}{1-G(x)}=\frac{1}{1-G(x)}\\ F(x)=1G(x)f[0]=1G(x)1
多项式求逆即可解决, O ( n l o g n ) O(nlogn) O(nlogn)

代码

分治FFT O ( n l o g 2 n ) O(nlog^2n) O(nlog2n)

#include<bits/stdc++.h>
#define ll long long 
#define M 400009//要开4倍! 
using namespace std;
int read(){
	int f=1,re=0;char ch;
	for(ch=getchar();!isdigit(ch)&&ch!='-';ch=getchar());
	if(ch=='-'){f=-1,ch=getchar();}
	for(;isdigit(ch);ch=getchar()) re=(re<<3)+(re<<1)+ch-'0';
	return re*f;
}
const int g=3;
const int mod=998244353; 
int rev[M];
int ksm(int a,int b){//快速幂 
	int ans=1;
	while(b){
		if(b&1) ans=(ll)ans*a%mod;
		a=(ll)a*a%mod;
		b>>=1;
	}return ans%mod;
}
void ntt(int *A,int lim,int type){//ntt
	for(int i=0;i<lim;i++) if(i<rev[i]) swap(A[i],A[rev[i]]);
	for(int mid=1;mid<lim;mid<<=1){
		int W=ksm(g,(mod-1)/(mid<<1));
		for(int R=mid<<1,j=0;j<lim;j+=R){
			int w=1;
			for(ll k=0;k<mid;k++,w=(ll)w*W%mod){
				int x=A[j+k],y=(ll)w*A[j+k+mid]%mod;
				A[j+k]=(x+y)%mod;
				A[j+mid+k]=(x-y+mod)%mod;
			}
		}
	}
	if(type==-1){
		reverse(A+1,A+lim);
        int inv=ksm(lim,mod-2);
        for(int i=0;i<lim;i++) A[i]=(ll)A[i]*inv%mod;
	}
}
int a[M],b[M],f[M],gg[M],n;
void cdq(int l,int r){
	if(l==r) return;
	int mid=(l+r)>>1;
	cdq(l,mid);
	int len=(r-l+1);
	int lim=1,lll=0;
	while(lim<len*2) lim<<=1,lll++;
	for(int i=0;i<lim;i++) rev[i]=(rev[i>>1]>>1)|((i&1)<<(lll-1)),a[i]=b[i]=0;
	for(int i=l;i<=mid;i++) a[i-l]=f[i];
	for(int i=1;i<=r-l;i++) b[i-1]=gg[i];
	ntt(a,lim,1),ntt(b,lim,1);
	for(int i=0;i<lim;i++) a[i]=(ll)a[i]*b[i]%mod;
	ntt(a,lim,-1);
	for(int i=mid+1;i<=r;i++) f[i]=(ll)(f[i]+a[i-l-1])%mod;
	cdq(mid+1,r);
}
signed main(){
	n=read();f[0]=1;
	for(int i=1;i<n;i++) gg[i]=read();
	cdq(0,n-1);
	for(int i=0;i<n;i++) printf("%d ",f[i]);
	printf("\n"); 
	return 0;
}

生成函数 O ( n l o g n ) O(nlogn) O(nlogn)

#include<bits/stdc++.h>
#define ll long long 
using namespace std;
const int g=3;
const int mod=998244353; 
const int M=2100009;
char s;
int read(){
	int f=1,re=0;char ch;
	for(ch=getchar();!isdigit(ch)&&ch!='-';ch=getchar());
	if(ch=='-'){f=-1,ch=getchar();}
	for(;isdigit(ch);ch=getchar()) re=(re<<3)+(re<<1)+ch-'0';
	return re*f;
}
int ksm(int a,int b){//快速幂 
	int ans=1;
	while(b){
		if(b&1) ans=(ll)ans*a%mod;
		a=(ll)a*a%mod;
		b>>=1;
	}return ans%mod;
}
int c[M],d[M],n,m,r[M],tmp[M],a[M],b[M],inv2=ksm(2,mod-2),dera[M],inva[M],lnb[M],lna[M],k,aa[M],bb[M],aaa[M],bbb[M],invb[M],F[M],cpy[M];
void ntt(int *A,int lim,int type){//ntt
	for(int i=0;i<lim;i++) if(i<r[i]) swap(A[i],A[r[i]]);
	for(int mid=1;mid<lim;mid<<=1){
		int W=ksm(g,(mod-1)/(mid<<1));
		for(int R=mid<<1,j=0;j<lim;j+=R){
			int w=1;
			for(ll k=0;k<mid;k++,w=(ll)w*W%mod){
				int x=A[j+k],y=(ll)w*A[j+k+mid]%mod;
				A[j+k]=(x+y)%mod;
				A[j+mid+k]=(x-y+mod)%mod;
			}
		}
	}
	if(type==-1){
		reverse(A+1,A+lim);
        int inv=ksm(lim,mod-2);
        for(int i=0;i<lim;i++) A[i]=(ll)A[i]*inv%mod;
	}
}
void getmul(int a[],int b[],int c[],int n,int m){//多项式乘法 
	int lim=1,l=0;
    while(lim<=n+m) lim<<=1,l++;
    for(int i=1;i<lim;i++) r[i]=(r[i>>1]>>1|(i&1)<<(l-1));
    for(int i=0;i<=n;i++) aa[i]=a[i];
    for(int i=0;i<=m;i++) bb[i]=b[i];
    ntt(aa,lim,1),ntt(bb,lim,1);
    for(int i=0;i<lim;i++){
        c[i]=(ll)aa[i]*bb[i]%mod;
        aa[i]=bb[i]=0;
    }ntt(c,lim,-1);
}
void getinv(int a[],int b[],int len){//多项式求逆 
	if(len==1){
		b[0]=ksm(a[0],mod-2);
		return;
	}getinv(a,b,(len+1)>>1);
	int lim=1,l=0;
	while(lim<len+len) lim<<=1,l++;
	for(int i=0;i<lim;i++) r[i]=(r[i>>1]>>1)|((i&1)<<(l-1));
	for(int i=0;i<len;i++) tmp[i]=a[i];
    for(int i=len;i<lim;i++) tmp[i]=0;
	ntt(tmp,lim,1),ntt(b,lim,1);
	for(int i=0;i<lim;i++) b[i]=(ll)b[i]*((2-(ll)tmp[i]*b[i]%mod+mod)%mod)%mod;
	ntt(b,lim,-1);
	for(int i=len;i<lim;i++) b[i]=0;
}
void getdiv(int a[],int b[],int c[],int d[],int n,int m){//多项式除法 
	for(int i=0;i<=n;i++) aaa[i]=a[i];
	for(int i=0;i<=m;i++) bbb[i]=b[i];
	reverse(aaa,aaa+n+1),reverse(bbb,bbb+m+1);
	getinv(bbb,invb,n-m+1);
	getmul(aaa,invb,c,n-m,n-m);
	reverse(c,c+n-m+1);
    reverse(aaa,aaa+n+1);reverse(bbb,bbb+m+1);
    getmul(c,bbb,d,n-m,m);
    for(int i=0;i<m;i++) d[i]=(aaa[i]-d[i]+mod)%mod;
    for(int i=0;i<=n;i++) aaa[i]=0;
	for(int i=0;i<=m;i++) bbb[i]=0;
}
void getsqrt(int a[],int b[],int len){//多项式开根 
	if(len==1) return (void)(b[0]=1);
	getsqrt(a,b,(len+1)>>1);
	for(int i=0;i<=(len<<1);i++) F[i]=0;
	getinv(b,F,len);
	int lim=1,l=0;
	while(lim<len+len) lim<<=1,l++;
	for(int i=0;i<lim;i++) r[i]=(r[i>>1]>>1)|((i&1)<<(l-1));
	for(int i=0;i<len;i++) cpy[i]=a[i];
    for(int i=len;i<lim;i++) cpy[i]=0;
	ntt(cpy,lim,1),ntt(b,lim,1),ntt(F,lim,1);
	for(int i=0;i<lim;i++) b[i]=((ll)((ll)(b[i]+(ll)cpy[i]*F[i]%mod)%mod)*inv2)%mod;
	ntt(b,lim,-1);
	for(int i=len;i<lim;i++) b[i]=0;
}
void getintegral(int a[],int b[],int len){//积分 
	for(int i=1;i<len;i++) b[i]=(ll)a[i-1]*ksm(i,mod-2)%mod;
	b[0]=0;
}
void getderivation(int a[],int b[],int len){//求导 
	for(int i=1;i<len;i++) b[i-1]=(ll)a[i]*i%mod;
	b[len-1]=0; 
}
void getln(int a[],int b[],int len){//多项式ln 
	getinv(a,inva,len);
	getderivation(a,dera,len);
	int lim=1,l=0;
	while(lim<len+len) lim<<=1,l++;
	for(int i=0;i<lim;i++) r[i]=(r[i>>1]>>1)|((i&1)<<(l-1));
	ntt(inva,lim,1),ntt(dera,lim,1);
	for(int i=0;i<lim;i++) inva[i]=(ll)inva[i]*dera[i]%mod;
	ntt(inva,lim,-1);
	getintegral(inva,b,len);
	for(int i=0;i<lim;i++) inva[i]=dera[i]=0;
}
void getexp(int a[],int b[],int len){//多项式exp 
	if(len==1) return (void)(b[0]=1);
	getexp(a,b,len>>1),getln(b,lnb,len);
	for(int i=0;i<len;i++) lnb[i]=(ll)(a[i]-lnb[i]+(i==0)+mod)%mod; 
	int lim=(len<<1);
	ntt(lnb,lim,1),ntt(b,lim,1);
	for(int i=0;i<lim;i++) b[i]=(ll)lnb[i]*b[i]%mod;
	ntt(b,lim,-1);
	for(int i=len;i<lim;i++) lnb[i]=b[i]=0;
}
void getksm(int a[],int b[],int len,int k){//多项式快速幂 
	getln(a,lna,len);
	for(int i=0;i<len;i++) lna[i]=(ll)k*lna[i]%mod;
	getexp(lna,b,len);
}
signed main(){
	n=read();a[0]=1;
	for(int i=1;i<n;i++) a[i]=-read();
	int lim=1;while(lim<=n) lim<<=1;
	getinv(a,b,lim);
	for(int i=0;i<n;i++) printf("%d ",b[i]);
	printf("\n");
	return 0;
}
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值