[다항식 알고리즘] 역 다항식 연구 노트

역 다항식

---

그리고 역수의 정수 다항식 같이 정의 \ (A (X) B (X) =. 1 \) 라고 \ (A (x)를, B는 (X) \) 서로 역수.

\ (A (X) \) 충분 조건 역수이다 \ ([X ^ 0]이 (X) \) 역수 존재한다.

지금 사용하는 방법에 대해 생각 \ (O (n \ 로그 N을 ) \) 를 계산하는 시간 \ (A (X) \) 곱셈 역의 :

승산을 고려 현재 전에 결정된 \ N- (\) 역 용어 다음 : \. (A (X) B (X) \ 당량 1 \ X ^ N-PMOD {} \)
\ [\ 수식 시작} {\ 시작 분할 {} A (x)의 B (x)는 -1 \ 당량 0 \ pmod를 X ^ {N} \\ \ 빅 (A (x)의 B (x)는 -1 \ 큰) ^ 2 \ equiv0 \ {X pmod에 2N ^ {}}이 A (\\ X) ^ 2B (X) ^ 2-2A (x)는 B (x)는 1 + \ 당량 0 \ {pmod를 X ^ {2N}} \\ 2A (X) B의 (X ) -A (x)는 ^ (b)는 ( X) ^ 2 \ equiv1 \ pmod에 {X ^ {2N}}이 A (x)의 B를 (\\ X) \ 빅 (2-A (x)의 B (x)는 \ 빅) 및 \ equiv1 \ {pmod를 X ^
{\\ 2N}} \ {단부 분할} \ {식 단부} \] 우리가 발견 할 수있을 것이다 \ (\ 개조 {X ^ { } 2N} \) 역에서 : \ (B (X) \ 빅 (2-A (X) B (X) \ 빅) \) , 반복 \ (O (\ 로그 N)를 \) 번 답을 얻을 수 있습니다.

시간 복잡도 : \ (T (N-) = T (\ FRAC N2) + O (N- \ N- 로그) = O (N- \ N- 로그) \)

코드 :

예 : P4238 [템플릿] 역 다항식

//Luogu-O2
#include <cmath>
#include <cstdio>
#include <cctype>
#include <cstring>
#include <algorithm>
#define rint register int
typedef long long ll;

#define Getchar (p1==p2&&(p2=(p1=In)+fread(In,1,1<<20,stdin),p1==p2)?EOF:*p1++)
char In[1<<20],*p1=In,*p2=In,Ch,Out[1<<20],*Outp=Out,St[25],*Tp=St;
inline int Getint(register int x=0)
{
    while(!isdigit(Ch=Getchar));
    for(;isdigit(Ch);Ch=Getchar)x=x*10+(Ch^48);
    return x;
}
inline void Putint(int x,char c)
{
    do *Tp++=x%10^48;while(x/=10);
    do *Outp++=*--Tp;while(Tp!=St);
    *Outp++=c;
}

const int p=998244353,g1=3,g2=(p+1)/3;

inline int Add(const int a,const int b){return (a+b)>=p?a+b-p:a+b;}
inline ll Pow(ll a,ll b,ll s=1){for(;b;b>>=1,a=a*a%p)if(b&1)s=s*a%p;return s;}

namespace Poly
{
    int r[1<<18];
    void Pre(const int n){for(rint i=1,l=log2(n);i<n;++i)r[i]=(r[i>>1]>>1)|((i&1)<<(l-1));}

    void NTT(const int n,int *A,const int g)
    {
        for(rint i=1;i<n;++i)if(i<r[i])std::swap(A[i],A[r[i]]);
        for(rint i=2,h=1;i<=n;i<<=1,h<<=1)
            for(rint j=0,Rs=Pow(g,(p-1)/i);j<n;j+=i)
                for(rint k=0,Rt=1;k<h;++k,Rt=(ll)Rt*Rs%p)
                {
                    int Tmp=(ll)A[j+h+k]*Rt%p;
                    A[j+h+k]=Add(A[j+k],p-Tmp),A[j+k]=Add(A[j+k],Tmp);
                }
        if(g==g2)for(rint i=0,In=Pow(n,p-2);i<n;++i)A[i]=(ll)A[i]*In%p;
    }

    int C[1<<18];
    void Inv(int n,int *A,int *B)
    {
        if(n==1){B[0]=Pow(A[0],p-2);return;}
        Inv(n>>1,A,B),memcpy(C,A,n*sizeof(int));
        Pre(n<<1),NTT(n<<1,C,g1),NTT(n<<1,B,g1);
        for(rint i=0;i<(n<<1);++i)B[i]=B[i]*(2-(ll)C[i]*B[i]%p+p)%p;
        NTT(n<<1,B,g2),memset(B+n,0,n*sizeof(int));
    }
}

int n,m,F[1<<18],G[1<<18];

int main()
{
    n=Getint()-1;
    for(rint i=0;i<=n;++i)F[i]=Getint();
    for(m=n,n=1;n<=m;n<<=1);
    Poly::Inv(n,F,G);
    for(rint i=0;i<=m;++i)Putint(G[i],i==m?'\n':' ');
    return fwrite(Out,1,Outp-Out,stdout),0;
}