洛谷P4238：【模板】多项式求逆

题目传送门：https://www.luogu.org/problemnew/show/P4238

---

题目分析：放个板子在blog上，以后复习的时候用。

注意每一次调用Poly_Rev()的时候都要重新计算一遍Rev数组，而且要先清0。

求$\mod x^m$下的逆元的时候，注意NTT的次数界要开到2m，因为$A(x)G^2(x)$的次数界加起来为$m+\frac{m}{2}+\frac{m}{2}-2=2m-2$。

这份板子看起来跑得还挺快的。

---

CODE：

#include<iostream>
#include<string>
#include<cstring>
#include<cmath>
#include<cstdio>
#include<cstdlib>
#include<stdio.h>
#include<algorithm>
using namespace std;

const int maxn=1000100;
const long long M=998244353;
const long long g=3;
typedef long long LL;

LL A[maxn];
LL B[maxn];

int Rev[maxn];
int N,Lg;

LL G[maxn];
LL F[maxn];

int n;

LL Pow(LL x,LL y)
{
    if (!y) return 1LL;
    LL temp=Pow(x,y>>1);
    temp=temp*temp%M;
    if (y&1) temp=temp*x%M;
    return temp;
}

void DFT(LL *a,int f)
{
    for (int i=0; i<N; i++)
        if (i<Rev[i]) swap(a[i],a[ Rev[i] ]);

    for (int len=2; len<=N; len<<=1)
    {
        int mid=(len>>1);
        LL e=Pow(g,(M-1)/len);
        if (f==-1) e=Pow(e,M-2);

        for (LL *p=a; p!=a+N; p+=len)
        {
            LL wn=1;
            for (int i=0; i<mid; i++)
            {
                LL temp=wn*p[mid+i]%M;
                p[mid+i]=(p[i]-temp+M)%M;
                p[i]=(p[i]+temp)%M;
                wn=wn*e%M;
            }
        }
    }
}

void NTT()
{
    for (int i=0; i<N; i++)
    {
        Rev[i]=0;
        for (int j=0; j<Lg; j++)
            if (i&(1<<j)) Rev[i]|=(1<<(Lg-j-1));
    }

    DFT(A,1);
    DFT(B,1);
    for (int i=0; i<N; i++) A[i]=A[i]*A[i]%M*B[i]%M;
    DFT(A,-1);

    LL inv=Pow(N,M-2);
    for (int i=0; i<N; i++) A[i]=A[i]*inv%M;
}

void Poly_Rev(int m)
{
    if (m==1) F[0]=Pow(G[0],M-2);
    else
    {
        Poly_Rev(m>>1);

        N=2*m,Lg=0;
        while ((1<<Lg)!=N) Lg++;
        for (int i=0; i<m; i++) A[i]=F[i],B[i]=G[i];
        for (int i=m; i<N; i++) A[i]=B[i]=0;
        NTT();

        for (int i=0; i<m; i++) F[i]=2LL*F[i]%M;
        for (int i=0; i<m; i++) F[i]=(F[i]-A[i]+M)%M;
    }
}

int main()
{
    freopen("4238.in","r",stdin);
    freopen("4238.out","w",stdout);

    scanf("%d",&n);
    for (int i=0; i<n; i++) scanf("%I64d",&G[i]);

    N=1;
    while (N<n) N<<=1;
    Poly_Rev(N);

    for (int i=0; i<n; i++) printf("%I64d ",F[i]);
    printf("\n");

    return 0;
}