「题意」给定(g[0]=1),(g[1~n-1])求序列(f[i]=sum_{j=1}^i f[i-j]*g[j] , iin[1,n-1],f[0]=1)。
「分析」分治处理区间[l,r],先递归求出[l,mid],在统计[l,mid]对[mid+1,r]的贡献,可以发现
[f[x]+=sum_{i=l}^{mid}f[i]*g[x-i] , xin[mid+1,r]
]
拿卷积算,令(A[0,mid-l])=(f[l,mid]), (B[l,r-l])=(g[1,r-l]) ,(B[0]=0),设(C[i])=(A[i]*B[i-j]), 那么
[f[x]+=sum f[i]*g[x-1]=sum A[i-l]*B[x-i]=C[x-l]
]
套上ntt
「代码」
#include <bits/stdc++.h>
using namespace std;
const int N=4e5+10;
const int P=998244353,G=3;
int n,lmt,l,rev[N];
int g[N],f[N],A[N],B[N];
int qpow(int x,int y) {
int c=1;
for(; y; y>>=1,x=1LL*x*x%P)
if(y&1) c=1LL*c*x%P;
return c;
}
void init(int len) {
for(lmt=1,l=0; lmt<len+len; lmt<<=1) l++;
for(int i=0; i<lmt; ++i) rev[i]=(rev[i>>1]>>1)|((i&1)<<(l-1));
}
void numberTheoreticTransform(int a[N],int tp) {
for(int i=0; i<lmt; ++i) if(i<rev[i]) swap(a[i],a[rev[i]]);
for(int m=1; m<lmt; m<<=1) {
long long wm=qpow(G,(P-1)/(m<<1));
if(tp==-1) wm=qpow(wm,P-2);
for(int i=0; i<lmt; i+=(m<<1)) {
long long w=1,tmp;
for(int j=0; j<m; ++j,w=w*wm%P) {
tmp=w*a[i+j+m]%P;
a[i+j+m]=(a[i+j]-tmp+P)%P;
a[i+j]=(a[i+j]+tmp)%P;
}
}
}
if(tp==-1) {
long long tmp=qpow(lmt,P-2);
for(int i=0; i<lmt; ++i) a[i]=tmp*a[i]%P;
}
}
void dfs(int l,int r) {
if(l==r) return;
int mid=(l+r)>>1;
dfs(l,mid);
init(r-l+1);
for(int i=0; i<lmt; ++i) A[i]=B[i]=0;
for(int i=l; i<=mid; ++i) A[i-l]=f[i];
for(int i=0; i<=r-l; ++i) B[i]=g[i];
numberTheoreticTransform(A,1);
numberTheoreticTransform(B,1);
for(int i=0; i<lmt; ++i) A[i]=1LL*A[i]*B[i]%P;
numberTheoreticTransform(A,-1);
for(int i=mid+1; i<=r; ++i) f[i]=(f[i]+A[i-l])%P;
dfs(mid+1,r);
}
int main() {
scanf("%d",&n);
for(int i=1; i<n; ++i) scanf("%d",g+i);
f[0]=1, dfs(0,n-1);
for(int i=0; i<n; ++i) printf("%d ",f[i]);
printf("
");
return 0;
}
ubuntu的中括号怎么是这个鬼玩意儿//