一我的的高三楼：多项式卷积，生成函数

Description：

一天的学习快要结束了，高三楼在晚自习的时候恢复了宁静。
不过，HSD 桑还有一些做业没有完成，他须要在这个晚自习写完。好比这道数学题：

一、给出一个数列，求它的前i项和$S_i$

HSD 桑擅长数学，很快就把这题秒了……
然而还有第二题：

二、若是把上一问的前n项和当作一个新数列，请求出它的i项和

 

看完第二题，还有第三题……HSD 桑已经预感到状况不妙了。
HSD 桑大体看了看题，发现有些规律。其实就是在求

次前缀和。若是咱们借用函数迭代的标记，就是在求 $S_n^{(k)}$
HSD 桑还有不少做业要写，请你帮助他完成这项做业。$mod \ 998244353$

$n \leq 100000 , k \leq 2^{63}$

时限：100ms

 

废话：

挺好的一道题。skyh一直在跟我说LNC给他颓了题解因而我也来作了。

然而我并不会因而打表找规律直接发现$O(n\ log\ n)$的式子而后就完了。

可是实际上丢了一个中间步骤$O(n\ log \ n log \  k)$

 

题解：

设咱们要求的$S_n^{(k)}$的生成函数$F_{(k)}(x)=\sum\limits_{i=0}^{n-1} S_{i+1}^{(k)} x^i$

而后考虑前缀和的过程，就是每一项都累加前面的全部项，也能够理解为每一个数都为后面的项作贡献

那么感受好像已经有卷积形式了，让咱们吧这个“贡献”方式也写成生成函数：

$G(x)=\sum\limits_{i=0}^{n-1}x^i$

而后按照贡献规则咱们能知道$F_{(k+1)}(x)=F_{(k)}(x) \times G(x)$

生成函数就是多项式了，这很卷积，因而咱们用快速幂的方式分别卷一下$F$和$G$就行。

这样的复杂度是$O(n\ log \ n log \  k)$的。时限故意卡掉了。

考虑这个$G$函数的具体意义：在$F$函数的表上往下走一步的同时往右走了若干步

根据范德蒙恒等式（我仍是喜欢叫枣树定理），组合数能够合并成一个

那么最后咱们能知道它的答案就是$f[i]=\sum\limits_{j=1}^{i} C_{k-1+i-j}^{k-1}  \times g[j]$

其中$g$是原数组，$f$是答案数组。

而后这是一个比较明显的卷积。至于怎么求组合数？

线性递推一乘一除就行了，k太大要及时取模。

 1 #include<cstdio>
 2 #define mod 998244353
 3 #define int long long
 4 int n,C[300000],x[300000],k,bin=1,rev[300000],INV;
 5 int pow(int b,int t,int a=1){for(;t;t>>=1,b=b*b%mod)if(t&1)a=a*b%mod;return a;}
 6 void NTT(int *a,int opt){
 7     for(int i=0;i<bin;++i)if(i<rev[i])a[i]^=a[rev[i]]^=a[i]^=a[rev[i]];
 8     for(int mid=1,wn;mid<bin;mid<<=1){
 9         wn=pow(3,(mod-1)/2/mid*opt+mod-1);
10         for(int i=0;i<bin;i+=mid<<1)
11             for(int j=0,w=1;j<mid;++j,w=w*wn%mod){
12                 int x=a[i+j],y=a[i+j+mid]*w%mod;
13                 a[i+j]=(x+y)%mod;a[i+j+mid]=(x-y+mod)%mod;
14             }
15     }
16     if(opt==-1)for(int i=0;i<bin;++i)a[i]=a[i]*INV%mod;
17 }
18 main(){
19     scanf("%lld%lld",&n,&k);k%=mod;
20     while(bin<=n<<1)bin<<=1; INV=pow(bin,mod-2);
21     for(int i=1;i<bin;++i)rev[i]=rev[i>>1]>>1|(i&1)*bin>>1;
22     for(int i=0;i<n;++i)scanf("%lld",&x[i]);
23     C[0]=1;
24     for(int i=1;i<n;++i)C[i]=C[i-1]*(k+i-1)%mod*pow(i,mod-2)%mod;
25     NTT(C,1);NTT(x,1);
26     for(int i=0;i<bin;++i)x[i]=x[i]*C[i]%mod;
27     NTT(x,-1);
28     for(int i=0;i<n;++i)printf("%lld\n",x[i]);
29 }

View Code