洛谷.4245.[模板]任意模数NTT(MTT/三模数NTT)

题目链接


三模数\(NTT\)

就是多模数\(NTT\)最后\(CRT\)一下...下面两篇讲的都挺明白的。
https://blog.csdn.net/kscla/article/details/79547242
https://blog.csdn.net/zhouyuheng2003/article/details/85561887

模数不是\(NTT\)模数,考虑用多个\(NTT\)模数分别卷积,最后\(CRT\)合并(由中国剩余定理,同余方程组在模\(M=\prod m_i\)的情况下解是唯一的)。
卷积后\(a_i\)的最大值是\((10^9)^2\times10^5=10^{23}\),所以需要选几个乘积\(>10^{23}\)的质数,比如\(m_1=998244353=2^{23}*119+1,\quad m_2=1004535809=2^{21}*479+1,\quad m_3=469762049=2^{26}*7+1\)。它们的原根都是\(3\)
最后我们能求出三个同余方程:\[x\equiv c_1\ (\mathbb{mod}\ m_1)\\x\equiv c_2\ (\mathbb{mod}\ m_2)\\x\equiv c_3\ (\mathbb{mod}\ m_3)\]

我们可以用\(CRT\)合并两个,就有\(x\equiv c_4\ (\mathbb{mod}\ m_1m_2)\)
\(x=am_1m_2+c_4\equiv c_3\ (\mathbb{mod}\ m_3)\),就可以算出\(a\equiv\frac{c_3-c_4}{m_1m_2}\ (\mathbb{mod}\ m_3)\),有了\(a\)代回去就算出\(x\)啦。
这样一共需要\(9\)\(DFT\),常数巨大。


拆系数\(FFT\)(\(MTT\))

\(FFT\)的问题在于精度,我们可以压缩值域。
\(m=\lceil\sqrt p\rceil,\ A_i=a\times m+b,\ B_i=c\times m+d\),那么\(h_{i+j}=\sum(a_im+b_i)(c_jm+d_j)=\sum a_ic_jm^2+(a_id_j+b_ic_j)m+b_id_j\),分别求出四项(中间两项放一块算),就只需要\(7\)\(DFT\)
为了方便可以令\(m=2^{15}=32768\)
这样\(FFT\)后的最大结果是\(10^5\times2^{30}=10^{14}\),注意取模。

注意std::sin的精度比sin高!因为至于依旧大所以必须要注意这个,还要开long double
还有个挺重要的优化是预处理单位根,预处理\(lim\)次单位根的\(0\sim lim-1\)次方,用\(\omega_i^k\)时就是\(\omega_{lim}^{k\times\frac{lim}{i}}\)(怎么都预处理的\(2lim\)次单位根啊...)。
注意预处理的时候不要像\(FFT\)里那样每次w=w*Wn,而是对每个单位根直接用欧拉公式算,这样精度误差会小非常多,就可以把long double替换成double了...(然后就可以快很多很多)
(嗯...误差问题都出在求单位根上了...)

现在\(7\)\(DFT\)已经挺快了...还可以优化,可以看毛啸的论文,或者这里以及这里我咕了。


三模数\(NTT\)

//3786ms    8.95MB
#include <cstdio>
#include <cctype>
#include <cstring>
#include <algorithm>
#define G 3
#define MOD(x,mod) x>=mod&&(x-=mod)
#define ADD(x,v,mod) (x+=v)>=mod&&(x-=mod)
#define gc() getchar()
#define MAXIN 500000
//#define gc() (SS==TT&&(TT=(SS=IN)+fread(IN,1,MAXIN,stdin),SS==TT)?EOF:*SS++)
typedef long long LL;
const int N=(1<<18)+3,Mod[]={998244353,469762049,1004535809};

int tA[N],tB[N],A[N],B[N],Ans[3][N],rev[N];
char IN[MAXIN],*SS=IN,*TT=IN;

inline int read()
{
    int now=0;register char c=gc();
    for(;!isdigit(c);c=gc());
    for(;isdigit(c);now=now*10+c-48,c=gc());
    return now;
}
inline int FP(int x,int k,const int mod)
{
    int t=1;
    for(; k; k>>=1,x=1ll*x*x%mod)
        if(k&1) t=1ll*t*x%mod;
    return t;
}
inline LL Mult(LL a,LL b,LL p)
{
    LL t=a*b-(LL)((long double)a/p*b+1e-8)*p;
    return t<0?t+p:t;
}
void NTT(int *a,int lim,int opt,const int mod)
{
    const int invG=FP(G,mod-2,mod);
    for(int i=1; i<lim; ++i) if(i<rev[i]) std::swap(a[i],a[rev[i]]);//&&
    for(int i=2; i<=lim; i<<=1)
    {
        int mid=i>>1,Wn=FP(~opt?G:invG,(mod-1)/i,mod);
        for(int j=0; j<lim; j+=i)
            for(int t,w=1,k=j; k<j+mid; ++k,w=1ll*w*Wn%mod)
                a[k+mid]=a[k]+mod-(t=1ll*a[k+mid]*w%mod), MOD(a[k+mid],mod),
                ADD(a[k],t,mod);
    }
    if(opt==-1) for(int i=0,inv=FP(lim,mod-2,mod); i<lim; ++i) a[i]=1ll*a[i]*inv%mod;
}
void Solve(const int lim,int *ans,const int mod)
{
    memcpy(A,tA,lim<<2), memcpy(B,tB,lim<<2);//不能只赋值到n!(清空后面的)
    NTT(A,lim,1,mod), NTT(B,lim,1,mod);
    for(int i=0; i<lim; ++i) A[i]=1ll*A[i]*B[i]%mod;
    NTT(A,lim,-1,mod);
    for(int i=0; i<lim; ++i) ans[i]=A[i];
}

int main()
{
    const int n=read(),m=read(),P=read();
    for(int i=0; i<=n; ++i) tA[i]=read(),tA[i]>=P&&(tA[i]%=P);
    for(int i=0; i<=m; ++i) tB[i]=read(),tB[i]>=P&&(tB[i]%=P);
    int lim=1,l=-1; while(lim<=n+m) lim<<=1,++l;
    for(int i=1; i<lim; ++i) rev[i]=(rev[i>>1]>>1)|((i&1)<<l);
    for(int i=0; i<3; ++i) Solve(lim,Ans[i],Mod[i]);
    #define m1 998244353ll
    #define m2 469762049ll
    #define m3 1004535809ll
    const LL M12=m1*m2,inv2=FP(m2,m1-2,m1),inv1=FP(m1,m2-2,m2),mul2=m2*inv2%M12,mul1=m1*inv1%M12,
            inv=FP(M12%m3,m3-2,m3),m12=M12%P;
    for(int i=0,t=n+m; i<=t; ++i)
    {
        LL c1=Ans[0][i],c2=Ans[1][i],c3=Ans[2][i],c4=Mult(c1,mul2,M12)+Mult(c2,mul1,M12);
        MOD(c4,M12);
        LL a=((c3-c4)%m3+m3)/*%m3*/*inv%m3;
        printf("%d ",(int)((a*m12+c4)%P));
    }

    return 0;
}

\(MTT\)

//1094ms    26.14MB
#include <cmath>
#include <cstdio>
#include <cctype>
#include <algorithm>
#define gc() getchar()
#define MAXIN 300000
//#define gc() (SS==TT&&(TT=(SS=IN)+fread(IN,1,MAXIN,stdin),SS==TT)?EOF:*SS++)
typedef long long LL;
typedef long double ld;
//#define double long double
const int N=(1<<18)+3;
const double PI=acos(-1);

int rev[N];
char IN[MAXIN],*SS=IN,*TT=IN;
struct Complex
{
    double x,y;
    Complex(double x=0,double y=0):x(x),y(y) {}
    inline Complex operator +(const Complex &a)const {return Complex(x+a.x, y+a.y);}
    inline Complex operator -(const Complex &a)const {return Complex(x-a.x, y-a.y);}
    inline Complex operator *(const Complex &a)const {return Complex(x*a.x-y*a.y, y*a.x+x*a.y);}
}A[N],B[N],C[N],D[N],W[N];

inline int read()
{
    int now=0; register char c=gc();
    for(;!isdigit(c);c=gc());
    for(;isdigit(c);now=now*10+c-'0',c=gc());
    return now;
}
void FFT(Complex *a,const int lim,const int opt)
{
    static Complex w[N];
    for(int i=1; i<lim; ++i) if(i<rev[i]) std::swap(a[i],a[rev[i]]);
    for(int i=2; i<=lim; i<<=1)
    {
        int mid=i>>1;
        if(~opt) for(int k=0,t=lim/i; k<mid; ++k) w[k]=W[k*t];
        else for(int k=0,t=lim/i; k<mid; ++k) w[k]=Complex(W[k*t].x,-W[k*t].y);
        for(int j=0; j<lim; j+=i)
        {
            Complex t;
            for(int k=j; k<j+mid; ++k)
                a[k+mid]=a[k]-(t=w[k-j]*a[k+mid]), a[k]=a[k]+t;
        }
    }
    if(opt==-1) for(int i=0; i<lim; ++i) a[i].x/=lim;
}

int main()
{
    const int n=read(),m=read(),P=read();
    for(int i=0,t; i<=n; ++i) t=read(),A[i].x=t>>15,B[i].x=t&32767;
    for(int i=0,t; i<=m; ++i) t=read(),C[i].x=t>>15,D[i].x=t&32767;

    int lim=1,l=-1; while(lim<=n+m) lim<<=1,++l;
    for(int i=0; i<lim; ++i) rev[i]=(rev[i>>1]>>1)|((i&1)<<l);
    for(int i=0,t=lim>>1; i<lim; ++i) W[i]=Complex(std::cos(i*PI/t),std::sin(i*PI/t));
//  const Complex Wn(std::cos(2.0*PI/lim),std::sin(2.0*PI/lim)); Complex w(1,0);
//  for(int i=0; i<lim; ++i) W[i]=w, w=w*Wn;//精度巨差 

    FFT(A,lim,1), FFT(B,lim,1), FFT(C,lim,1), FFT(D,lim,1);
    for(int i=0; i<lim; ++i)
    {
        const Complex a=A[i],b=B[i],c=C[i],d=D[i];
        A[i]=a*c, B[i]=a*d+b*c, D[i]=b*d;
    }
    FFT(A,lim,-1), FFT(B,lim,-1), FFT(D,lim,-1);
    for(int i=0,t=n+m; i<=t; ++i)
    {
        LL res=((LL(A[i].x+0.5))%P<<30)+((LL(B[i].x+0.5))%P<<15)+(LL(D[i].x+0.5));
        printf("%d ",(int)(res%P));
    }

    return 0;
}

猜你喜欢

转载自www.cnblogs.com/SovietPower/p/10546764.html