CF923E Perpetual Subtraction

生成函数好题!

搬一手铃悬的题解(侵删)

现在只需要考虑怎么求出g和逆变换即可,其实也就是对函数F(x)求F(x+1)和F(x-1)。

直接二项式定理展开发现是个卷积的形式,大力NTT即可。

#include<bits/stdc++.h>
#define N 440000
#define eps 1e-7
#define inf 1e9+7
#define db double
#define ll long long
#define ldb long double
using namespace std;
inline int read()
{
    char ch=0;
    int x=0,flag=1;
    while(!isdigit(ch)){ch=getchar();if(ch=='-')flag=-1;}
    while(isdigit(ch)){x=(x<<3)+(x<<1)+ch-'0';ch=getchar();}
    return x*flag;
}
const int d=3,mo=998244353;
int ksm(int x,int k)
{
    int ans=1;
    while(k)
    {
        if(k&1)ans=1ll*ans*x%mo;
        k>>=1;x=1ll*x*x%mo;
    }
    return ans;
}
int rev[N];
void ntt(int *f,int n,int flag)
{
    for(int i=0;i<n;i++)
    {
        rev[i]=(rev[i>>1]>>1)+(i&1)*(n>>1);
        if(i<rev[i])swap(f[i],f[rev[i]]);
    }
    for(int k=2,kk=1;k<=n;k<<=1,kk<<=1)
    {
        int wn=ksm(d,(mo-1)/k);
        if(flag==-1)wn=ksm(wn,mo-2);
        for(int i=0;i<n;i+=k)
        for(int j=0,w=1;j<kk;j++,w=1ll*w*wn%mo)
        {
            int t=1ll*w*f[i+j+kk]%mo;
            f[i+j+kk]=(f[i+j]-t)%mo;
            f[i+j]=(f[i+j]+t)%mo;
        }
    }
    if(flag==-1)
    {
        int k=ksm(n,mo-2);
        for(int i=0;i<n;i++)f[i]=1ll*f[i]*k%mo;
    }
}
int a[N],b[N];
void mul(int len)
{
    ntt(a,len,+1);ntt(b,len,+1);
    for(int i=0;i<len;i++)a[i]=1ll*a[i]*b[i]%mo;
    ntt(a,len,-1);
}
int n,m,len,f[N],g[N],fac[N],vac[N];
int main()
{
    n=read();ll t;cin>>t;m=(t%(mo-1));len=1;
    while(len<2*(n+1))len<<=1;
    for(int i=0;i<=n;i++)f[i]=read();
    fac[0]=vac[0]=1;
    for(int i=1;i<=len;i++)fac[i]=1ll*fac[i-1]*i%mo;
    vac[len]=ksm(fac[len],mo-2);
    for(int i=len-1;i>=1;i--)vac[i]=1ll*vac[i+1]*(i+1)%mo;
    //get g(x)=f(x+1)
    for(int i=0;i<=n;i++)a[i]=1ll*f[i]*fac[i]%mo,b[i]=vac[i];
    for(int i=n+1;i<len;i++)a[i]=b[i]=0;
    reverse(a,a+n+1);mul(len);
    for(int i=0;i<=n;i++)g[i]=1ll*vac[i]*a[n-i]%mo;
    //solve get g*(x)
    for(int i=0;i<=n;i++)g[i]=1ll*ksm(ksm(i+1,m),mo-2)*g[i]%mo;
    //get f*(x)=g(x-1)
    for(int i=0;i<=n;i++)a[i]=1ll*g[i]*fac[i]%mo,b[i]=1ll*ksm(-1,i)*vac[i]%mo;
    for(int i=n+1;i<len;i++)a[i]=b[i]=0;
    reverse(a,a+n+1);mul(len);
    for(int i=0;i<=n;i++)f[i]=1ll*vac[i]*a[n-i]%mo;
    //print f(x) 
    for(int i=0;i<=n;i++)printf("%d ",(f[i]%mo+mo)%mo);
    return 0;
}

猜你喜欢

转载自www.cnblogs.com/Creed-qwq/p/10787950.html