码迷,mamicode.com
首页 > 其他好文 > 详细

多项式全家桶

时间:2019-12-21 20:18:12      阅读:55      评论:0      收藏:0      [点我收藏+]

标签:etl   cstring   read   cst   str   printf   namespace   reverse   wap   

#include<iostream>
#include<cstring>
#include<cstdio>
#include<algorithm>
#define mod 998244353
using namespace std;
int n,a[600100],b[600100],ina[600100],lna[600100],lnc[600100],sqa[600100],sqb[600100],exa[600100],exb[600100],poa[600100],pob[600100],pos[600100];
int read()
{
    int aa=0,bb=1;char cc=getchar();
    while(cc>9||cc<0){if(cc==-) bb=-1;cc=getchar();}
    while(cc>=0&&cc<=9){aa=(aa<<3)+(aa<<1)+(cc^0);cc=getchar();}
    return aa*bb;
}
int quick(int x,int p)
{
    int as=1;
    while(p){
        if(p&1) as=1ll*as*x%mod;
        x=1ll*x*x%mod;p>>=1;
    }
    return as;
}
void ntt(int *a,int len,int opt)
{
    for(int i=0;i<len;i++) if(i<pos[i]) swap(a[i],a[pos[i]]);
    for(int i=1;i<len;i<<=1){
        int step=i<<1,wn=quick(3,(mod-1)/step);
        for(int j=0;j<len;j+=step){
            int w=1;
            for(int k=0;k<i;k++,w=1ll*w*wn%mod){
                int x=a[j+k],y=1ll*a[j+k+i]*w%mod;
                a[j+k]=(x+y)%mod;a[j+k+i]=(x-y+mod)%mod;
            }
        }
    }
    if(opt){
        reverse(a+1,a+len);int inv=quick(len,mod-2);
        for(int i=0;i<len;i++) a[i]=1ll*a[i]*inv%mod;
    }
}
void getderiv(int n,int *a,int *b)
{
    for(int i=1;i<n;i++) b[i-1]=1ll*a[i]*i%mod;
    b[n-1]=0;
}
void getinteg(int n,int *a,int *b)
{
    for(int i=1;i<n;i++) b[i]=1ll*a[i-1]*quick(i,mod-2)%mod;
    b[0]=0;
}
void getinv(int n,int *a,int *b)
{
    if(n==1){b[0]=quick(a[0],mod-2);return;}
    getinv((n+1)>>1,a,b);
    int lenth=1,bit=0;
    while(lenth<=n+n+n) lenth<<=1,bit++;
    for(int i=0;i<lenth;i++) pos[i]=(pos[i>>1]>>1)|((i&1)<<bit-1);
    for(int i=0;i<n;i++) ina[i]=a[i];
    for(int i=n;i<lenth;i++) ina[i]=0;
    ntt(ina,lenth,0);ntt(b,lenth,0);
    for(int i=0;i<lenth;i++) b[i]=(2ll*b[i]%mod-1ll*ina[i]*b[i]%mod*b[i]%mod+mod)%mod;
    ntt(b,lenth,1);
    for(int i=n;i<lenth;i++) b[i]=0;
}
void getln(int n,int *a,int *b)
{
    getderiv(n,a,lna);
    getinv(n,a,lnc);
    int lenth=1,bit=0;
    while(lenth<=n+n) lenth<<=1,bit++;
    for(int i=0;i<lenth;i++) pos[i]=(pos[i>>1]>>1)|((i&1)<<bit-1);
    ntt(lna,lenth,0);ntt(lnc,lenth,0);
    for(int i=0;i<lenth;i++) lna[i]=1ll*lna[i]*lnc[i]%mod;
    ntt(lna,lenth,1);getinteg(n,lna,b);
    for(int i=0;i<lenth;i++) lna[i]=0,lnc[i]=0;
}
void getsqrt(int n,int *a,int *b)
{
    if(n==1){b[0]=1;return;}
    getsqrt((n+1)>>1,a,b);
    getinv(n,b,sqb);
    int lenth=1,bit=0;
    while(lenth<=n+n+n) lenth<<=1,bit++;
    for(int i=0;i<lenth;i++) pos[i]=(pos[i>>1]>>1)|((i&1)<<bit-1);
    for(int i=0;i<n;i++) sqa[i]=a[i];
    for(int i=n;i<lenth;i++) sqa[i]=0;
    ntt(b,lenth,0);ntt(sqa,lenth,0);ntt(sqb,lenth,0);int inv=quick(2,mod-2);
    for(int i=0;i<lenth;i++) b[i]=(1ll*b[i]*b[i]%mod+sqa[i])%mod*sqb[i]%mod*inv%mod;
    ntt(b,lenth,1);
    for(int i=n;i<lenth;i++) b[i]=0;
    for(int i=0;i<lenth;i++) sqb[i]=0;
}
void getexp(int n,int *a,int *b)
{
    if(n==1){b[0]=1;return;}
    getexp((n+1)>>1,a,b);
    getln(n,b,exb);
    int lenth=1,bit=0;
    while(lenth<=n+n) lenth<<=1,bit++;
    for(int i=1;i<lenth;i++) pos[i]=(pos[i>>1]>>1)|((i&1)<<bit-1);
    for(int i=0;i<n;i++) exb[i]=(a[i]-exb[i]+mod)%mod;exb[0]++;
    for(int i=n;i<lenth;i++) exb[i]=0;
    ntt(exb,lenth,0);ntt(b,lenth,0);
    for(int i=0;i<lenth;i++) b[i]=1ll*b[i]*exb[i]%mod;
    ntt(b,lenth,1);
    for(int i=n;i<lenth;i++) b[i]=0;
}
void getpow(int n,int *a,int *b,int k)
{
    getln(n,a,poa);
    for(int i=0;i<n;i++) pob[i]=1ll*poa[i]*k%mod;
    getexp(n,pob,b);
    int lenth=1,bit=0;
    while(lenth<=n) lenth<<=1,bit++;
    for(int i=n;i<lenth;i++) b[i]=0;
    for(int i=0;i<lenth;i++) poa[i]=0,pob[i]=0;
}
int main()
{
    n=read();
    for(int i=0;i<n;i++) a[i]=read();
    getsqrt(n,a,b);
    for(int i=0;i<n;i++) printf("%d ",b[i]);
    return 0;
}

 

啥都不会只会板子(现在连板子都不会了

多项式全家桶

标签:etl   cstring   read   cst   str   printf   namespace   reverse   wap   

原文地址:https://www.cnblogs.com/jrf123/p/12077945.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!