HDU 6061 RXD and functions（NTT）

Description

Input
多组用例，每组用例首先输入一整数n表示f的次数，之后输入n+1个整数c[i]表示f的系数，然后输入一整数m，之后输入m个整数a[i]，以文件尾结束输入
(n<=1e5,0<=a[i],c[i]<998244353,sum{m}<=1e5)
Output
对于每组用例，输出b[0]~b[n]
Sample Input
2
0 0 1
1
1
Sample Output
1 998244351 1
Solution

Code

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int maxbit=18,maxlen=1<<maxbit,mod=998244353,g=3,maxn=100001;
int wn[maxlen],inv2[maxbit+1],fact[maxn],inv[maxn];
int mod_pow(int a,int b)
{
    int ans=1;
    while(b)
    {
        if(b&1)ans=(ll)ans*a%mod;
        a=(ll)a*a%mod;
        b>>=1;
    }
    return ans;
}
void init()
{
    wn[0]=1,wn[1]=mod_pow(g,(mod-1)>>maxbit);
    for(int i=2;i<maxlen;i++)wn[i]=(ll)wn[i-1]*wn[1]%mod;
    inv2[0]=1,inv2[1]=(mod+1)/2;
    for(int i=2;i<=maxbit;i++)inv2[i]=(ll)inv2[i-1]*inv2[1]%mod;//预处理2^i的逆元
    fact[0]=1;
    for(int i=1;i<maxn;i++)fact[i]=(ll)i*fact[i-1]%mod;//求i!
    inv[0]=inv[1]=1;
    for(int i=2;i<maxn;i++)inv[i]=mod-(int)(mod/i*(ll)inv[mod%i]%mod);//线性求1~n逆元
    for(int i=1;i<maxn;i++)inv[i]=(ll)inv[i-1]*inv[i]%mod;//预处理i!的逆元
}
void ntt(int *x,int len,int sta)
{
    for(int i=0,j=0;i<len;i++)
    {
        if(i>j)swap(x[i],x[j]);
        for(int l=len>>1;(j^=l)<l;l>>=1);
    }
    for(int i=1,d=1;d<len;i++,d<<=1)
        for(int j=0;j<len;j+=d<<1)
            for(int k=0;k<d;k++)
            {
                int t=(ll)wn[(maxlen>>i)*k]*x[j+k+d]%mod;
                x[j+d+k]=x[j+k]-t<0?x[j+k]-t+mod:x[j+k]-t;
                x[j+k]=x[j+k]+t>=mod?x[j+k]+t-mod:x[j+k]+t;
            }
    if(sta==-1)
    {
        reverse(x+1,x+len);
        int bitlen=0;
        while((1<<bitlen)<len)bitlen++;
        int val=inv2[bitlen];
        for(int i=0;i<len;i++)x[i]=(ll)x[i]*val%mod;
    }
}
void NTT(int *a,int *b,int len1,int len2)
{
    int len=1;
    while(len<len1+len2)len<<=1;
    for(int i=len1;i<len;i++)a[i]=0;
    for(int i=len2;i<len;i++)b[i]=0;
    ntt(a,len,1),ntt(b,len,1);
    for(int i=0;i<len;i++)a[i]=(ll)a[i]*b[i]%mod;
    ntt(a,len,-1);
}
int n,m,c[maxn],a;
int A[maxlen],B[maxlen];
int main()
{
    init();
    while(~scanf("%d",&n))
    {
        for(int i=0;i<=n;i++)scanf("%d",&c[i]);
        scanf("%d",&m);
        a=0;
        for(int i=0;i<m;i++)
        {
            int t;
            scanf("%d",&t);
            a+=t;
            if(a>=mod)a-=mod;
        }
        int temp=1;
        A[0]=1;
        for(int i=1;i<=n;i++)
        {
            temp=(ll)temp*a%mod;
            temp=mod-temp;
            A[i]=(ll)temp*inv[i]%mod;
        }
        for(int i=0;i<=n;i++)B[i]=(ll)c[n-i]*fact[n-i]%mod;
        NTT(A,B,n+1,n+1);
        for(int i=0;i<=n;i++)A[i]=(ll)A[i]*inv[n-i]%mod;
        for(int i=0;i<=n;i++)printf("%d ",A[n-i]);
        printf("\n");
    }
}