LOJ #6261 一个人的高三楼

生成函数和组合数学的灵活应用

LOJ #6261

题意：求一个数列的$ k$次前缀和

---

$ Solution:$

我们对原数列$ a$建生成函数$ A=\sum\limits_{i=0}^{n-1} a_ix^i $

求一次前缀和相当于将$ A$卷上生成函数$ B=\sum\limits_{i=0}^{n-1}x^i \pmod{x^n}$

即我们要求的就是$ A·B^{k-1} \pmod{x^n}$

直接快速幂是$ log^2$的，但是生成函数$ B$有一些巧妙的性质：

$ B^k(x)$的意义是选$ k$个自然数使得和为$ x$

可以通过插板法得知$ B^k(x)$=$ C_{x+k}^x$

然后求出$ A,B$之后$ NTT$卷积即可

时间复杂度：$ O(n \ log \ n)$

---

$ my \ code:$

#include<ctime>
#include<cmath>
#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
#include<queue>
#define p 998244353
#define rt register int
#define ll long long
using namespace std;
inline ll read(){
    ll x = ; char zf = ; char ch = getchar();
    while (ch != '-' && !isdigit(ch)) ch = getchar();
    if (ch == '-') zf = -, ch = getchar();
    while (isdigit(ch)) x = x *  + ch - '', ch = getchar(); return x * zf;
}
void write(ll y){if(y<)putchar('-'),y=-y;if(y>)write(y/);putchar(y%+);}
void writeln(const ll y){write(y);putchar('\n');}
int i,j,k,n,x,y,z,cnt,lim,m;
vector<int>A,B,R;
int inv[];
int ksm(int x,int y){
    int ans=;
    for(rt i=y;i;i>>=,x=1ll*x*x%p)if(i&)ans=1ll*ans*x%p;
    return ans;
}
void init(){
    lim=;while(lim<=n+n)lim*=;
    A.resize(lim);B.resize(lim);R.resize(lim);
    for(rt i=;i<lim;i++)R[i]=(R[i>>]>>)|(i&)*(lim>>);
    for(rt i=;i<n;i++)A[i]=read();
    inv[]=inv[]=;B[]=;
    for(rt i=;i<=n;i++)inv[i]=1ll*inv[p%i]*(p-p/i)%p;
    for(rt i=;i<n;i++)B[i]=1ll*(i+m)*B[i-]%p*inv[i]%p;
}
void NTT(int n,vector<int>&A,int fla){
    for(rt i=;i<n;i++)if(i>R[i])swap(A[i],A[R[i]]);
    for(rt i=;i<n;i<<=){
        int w=ksm(,(p-)//i);
        for(rt j=;j<n;j+=i<<){
            int K=;
            for(rt k=;k<i;k++,K=1ll*K*w%p){
                int x=A[j+k],y=1ll*K*A[i+j+k]%p;
                A[j+k]=(x+y)%p,A[i+j+k]=(x-y)%p;
            }
        }
    }
    if(fla==-){
        reverse(A.begin()+,A.end());int invn=ksm(n,p-);
        for(rt i=;i<n;i++)A[i]=1ll*A[i]*invn%p;
    }
}
int main(){
    n=read();m=(read()-)%p;init();
    NTT(lim,A,);NTT(lim,B,);
    for(rt i=;i<lim;i++)A[i]=1ll*A[i]*B[i]%p;
    NTT(lim,A,-);
    for(rt i=;i<n;i++)writeln((A[i]+p)%p);
    return ;
}