多项式模板合集

2020-01-27 09:57:49 阅读：207 来源： 互联网

在绞尽脑汁学习过后记下的一些笔记和代码……
我也只是半懂，\(so\) 这篇文章对初学者可能不太友好。

好的参考资料(来自 \(Miskcoo's\) \(space\) )：
从多项式乘法到快速傅里叶变换
 多项式求逆元
 多项式除法及求模
 多项式的多点求值与快速插值
 牛顿迭代法在多项式运算的应用

对于下述代码，“前置”代码：

#define P 998244353
int Pow_mod(int x,int y){
    int ret=1;
    while(y){
        if(y&1) ret=1ll*ret*x%P;
        x=1ll*x*x%P;
        y>>=1;
    }
    return ret;
}
int Plus(int x,int y) { return x+y>=P ? x+y-P : x+y; }
int Minus(int x,int y) { return x>=y ? x-y : x-y+P; }

FFT/NTT

int l,r[N],X[N];
void ntt(int *A,int ty){
    for(int i=0;i<l;i++) X[r[i]]=A[i];
    for(int i=0;i<l;i++) A[i]=X[i];
    for(int i=2;i<=l;i<<=1){
        int wn=Pow_mod(3,(P-1)/i);
        if(ty==-1) wn=Pow_mod(wn,P-2);
        for(int j=0;j<l;j+=i){
            int w=1;
            for(int k=j;k<j+i/2;k++){
                int t=1ll*w*A[k+i/2]%P;
                A[k+i/2]=Minus(A[k],t);
                A[k]=Plus(A[k],t);
                w=1ll*w*wn%P;
            }
        }
    } 
    if(ty==1) return;
    int Inv=Pow_mod(l,P-2);
    for(int i=0;i<l;i++) A[i]=1ll*A[i]*Inv%P;
}

多项式求逆

int C[N];
void inv(int *A,int *B,int x){
    if(x==1) { B[0]=Pow_mod(A[0],P-2); return; }
    inv(A,B,(x+1)>>1);
    
    l=1;
    while(l<2*x) l<<=1;
    for(int i=1;i<l;i++) r[i]=(r[i>>1]>>1)|((i&1)*(l>>1));
    
    for(int i=0;i<x;i++) C[i]=A[i];
    for(int i=x;i<l;i++) C[i]=0;
    ntt(C,1); ntt(B,1);
    for(int i=0;i<l;i++) B[i]=Minus(2ll*B[i]%P,1ll*C[i]*B[i]%P*B[i]%P);
    ntt(B,-1);
    
    for(int i=x;i<l;i++) B[i]=0;
}

多项式除法&取模

void div(int *A,int *B,int *D,int *R){ //A[N]有n项,B[N]有m项。A[x]=B[x]*D[x]+R[x]
    int Ar[N],Br[N],Dr[N];
    for(int i=0;i<n;i++) Ar[i]=A[n-i-1];
    for(int i=0;i<m;i++) Br[i]=B[m-i-1];
    inv(Br,Dr,n-m+1);
    l=1;
    while(l<2*n) l<<=1;
    for(int i=1;i<l;i++) r[i]=(r[i>>1]>>1)|((i&1)*(l>>1));
    ntt(Dr,1); ntt(Ar,1);
    for(int i=0;i<l;i++) Dr[i]=1ll*Dr[i]*Ar[i]%P;
    ntt(Dr,-1);
    for(int i=0;i<=n-m;i++) D[i]=Dr[n-m-i];
    
    for(int i=0;i<m;i++) Br[i]=B[i];
    ntt(Br,1); ntt(D,1);
    for(int i=0;i<l;i++) Br[i]=1ll*Br[i]*D[i]%P;
    ntt(Br,-1); ntt(D,-1);
    for(int i=0;i<m-1;i++) R[i]=Minus(A[i],Br[i]);
}

多项式多点求值

多项式取log

多项式exp

FWT

标签：return,int,多项式,void,ntt,ret,合集,模板
来源： https://www.cnblogs.com/lindalee/p/12222196.html

本站声明： 1. iCode9 技术分享网（下文简称本站）提供的所有内容，仅供技术学习、探讨和分享；
2. 关于本站的所有留言、评论、转载及引用，纯属内容发起人的个人观点，与本站观点和立场无关；
3. 关于本站的所有言论和文字，纯属内容发起人的个人观点，与本站观点和立场无关；
4. 本站文章均是网友提供，不完全保证技术分享内容的完整性、准确性、时效性、风险性和版权归属；如您发现该文章侵犯了您的权益，可联系我们第一时间进行删除；
5. 本站为非盈利性的个人网站，所有内容不会用来进行牟利，也不会利用任何形式的广告来间接获益，纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。

ICode9