【模板】多项式指数函数(多项式 exp)

#include<bits/stdc++.h>
#define ll long long
using namespace std;
const int N=4e5+77,mod=998244353;
const ll g=3;
int r[N];
ll a[N],b[N],c[N],ia[N],da[N],_inv[N],f[N];
ll power(ll x,ll t)
{
    ll b=1;
    while(t)
    {
        if(t&1) b=b*x%mod; x=x*x%mod; t>>=1;
    }
    return b;
}
void dft(ll *a,int n,int f)
{
    for(int i=1; i<=n; i++) if(i<r[i]) swap(a[i],a[r[i]]);
    for(int i=2; i<=n; i<<=1)
    {
        int now=i>>1; ll wn=power(g,(mod-1)/i);
        if(f==-1) wn=power(wn,mod-2);
        for(int j=0; j<n; j+=i)
        {
            ll x,y,w=1;
            for(int k=j; k<j+now; k++,w=w*wn%mod)
            {
                x=a[k],y=w*a[k+now]%mod;
                a[k]=(x+y)%mod,a[k+now]=(x-y+mod)%mod;
            }
        }
    }
    if(f==-1) for(int i=0; i<n; i++) a[i]=a[i]*power(n,mod-2)%mod;
}
void inv(int n,ll *a,ll *b)
{
    if(n==1)
    {
        b[0]=power(a[0],mod-2); return;
    }
    inv((n+1)>>1,a,b);
    int len=1,lg=0;
    while(len<n*2) len<<=1,lg++;
    for(int i=1; i<=len; i++) r[i]=((r[i>>1]>>1)|((i&1)<<(lg-1)));
    for(int i=0; i<n; i++) c[i]=a[i]; for(int i=n; i<len; i++) c[i]=0;
    dft(c,len,1),dft(b,len,1);
    for(int i=0; i<len; i++) b[i]=(2-c[i]*b[i]%mod+mod)%mod*b[i]%mod;
    dft(b,len,-1);
    for(int i=n; i<len; i++) b[i]=0;
}
void dev(int n,ll *a,ll *b)
{
    for(int i=1; i<n; i++) b[i-1]=i*a[i]%mod; b[n-1]=0;
}
void invdev(int n,ll *a,ll *b)
{
    for(int i=1; i<n; i++) b[i]=a[i-1]*_inv[i]%mod; b[0]=0;
}
void ln(int n,ll *a,ll *b)
{
    dev(n,a,da); inv(n,a,ia);
    int len=1,lg=0;
    while(len<n*2) len<<=1,lg++;
    for(int i=1; i<=len; i++) r[i]=((r[i>>1]>>1)|((i&1)<<(lg-1)));
    dft(da,len,1); dft(ia,len,1);
    for(int i=0; i<len; i++) c[i]=da[i]*ia[i]%mod;
    dft(c,len,-1);
    invdev(len,c,b);
    for(int i=0; i<len; i++) da[i]=ia[i]=0;
}
void exp(int n,ll *a,ll *b)
{
    if(n==1)
    {
        b[0]=1; return;
    }
    exp(n>>1,a,b); 
    ln(n,b,f);
    f[0]=(a[0]+1-f[0]+mod)%mod;
    for(int i=1; i<n; i++) f[i]=(a[i]-f[i]+mod)%mod;
    int len=1,lg=0;
    while(len<n<<1) len<<=1,lg++;
    for(int i=1; i<=len; i++) r[i]=(r[i>>1]>>1)|((i&1)<<(lg-1));
    dft(b,len,1),dft(f,len,1);
    for(int i=0; i<len; i++) b[i]=b[i]*f[i]%mod;
    dft(b,len,-1);
    for(int i=n; i<len; i++) b[i]=f[i]=0;
//    for(int i=0; i<n; i++) printf("%lld ",b[i]); printf("\n");
}
int main()
{
    int n;
    scanf("%d",&n);
    for(int i=0; i<n; i++) scanf("%lld",&a[i]);
    int len=1;
    while(len<=n) len<<=1;
    _inv[1]=1;
    for(int i=2; i<=len; i++) _inv[i]=(mod-(mod/i))*_inv[mod%i]%mod;
    exp(len,a,b);
    for(int i=0; i<n; i++) printf("%lld ",b[i]);
}

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值