Luogu P5667 拉格朗日插值2

Link
根据Lagrange多项式,(f(m+i)=sumlimits_{j=0}^nf(j)prodlimits_{k e j}frac{m+i-k}{j-k}=frac{(m+i)!}{(m-n+i-1)!}sumlimits_{j=0}^nfrac{(-1)^{n-j}f(j)}{j!(n-j)!m+i-j})
平移之后卷积一下就行了。

#include<cstdio>
#include<cctype>
#include<cstring>
#include<algorithm>
const int N=524288,P=998244353;
int n,m,lim,ifac[N],minv[N],mfac[N],mifac[N],rev[N],w[N],f[N],g[N];
int read(){int x=0,c=getchar();while(isspace(c))c=getchar();while(isdigit(c))(x*=10)+=c&15,c=getchar();return x;}
void inc(int&a,int b){a+=b-P,a+=a>>31&P;}
void dec(int&a,int b){a-=b,a+=a>>31&P;}
int mul(int a,int b){return 1ll*a*b%P;}
int pow(int a,int k){int r=1;for(;k;k>>=1,a=mul(a,a))if(k&1)r=mul(a,r);return r;}
void init()
{
    int p=(lim=1<<(32-__builtin_clz(3*n)))>>1,g=pow(3,(P-1)/lim);
    w[p]=ifac[0]=mfac[0]=minv[0]=mifac[0]=1;
    for(int i=1;i<lim;++i) rev[i]=(rev[i>>1]>>1)|(i&1? p:0);
    for(int i=p+1;i<lim;++i) w[i]=mul(w[i-1],g);
    for(int i=p-1;i;--i) w[i]=w[i<<1];
    for(int i=1;i<=n;++i) ifac[i]=mul(ifac[i-1],pow(i,P-2));
    for(int i=1;i<=2*n+1;++i) mfac[i]=mul(mfac[i-1],m-n+i-1),mifac[i]=mul(mifac[i-1],minv[i]=pow(m-n+i-1,P-2));
}
void NTT(int*a,int f)
{
    if(!~f) std::reverse(a+1,a+lim);
    for(int i=1;i<lim;++i) if(i<rev[i]) std::swap(a[i],a[rev[i]]);
    for(int i=1;i<lim;i<<=1) for(int j=0,d=i<<1;j<lim;j+=d) for(int k=0,x;k<i;++k) x=mul(a[i+j+k],w[i+k]),dec(a[i+j+k]=a[j+k],x),inc(a[j+k],x);
    if(!~f) for(int i=0,x=P-(P-1)/lim;i<lim;++i) a[i]=mul(a[i],x);
}
int main()
{
    n=read(),m=read(),init();
    for(int i=0;i<=n;++i) if(f[i]=mul(mul(ifac[i],ifac[n-i]),read()),(n-i)&1) f[i]=P-f[i];
    for(int i=0;i<=n*2;++i) g[i]=minv[i+1];
    NTT(f,1),NTT(g,1);
    for(int i=0;i<lim;++i) f[i]=mul(f[i],g[i]);
    NTT(f,-1);
    for(int i=n;i<=n*2;++i) printf("%d ",mul(mul(mfac[i+1],mifac[i-n]),f[i]));
}
原文地址:https://www.cnblogs.com/cjoierShiina-Mashiro/p/12715104.html