洛谷4245：【模板】任意模数NTT——题解

https://www.luogu.org/problemnew/show/P4245

给两个多项式，求其乘积，每个系数对p取模。

参考：

代码与部分理解参考https://www.luogu.org/blog/yhzq/solution-p4245

NTT常用模数https://blog.csdn.net/hnust_xx/article/details/76572828

一些有关NTT讲解的东西。

————————————

NTT作用和DFT相同，只是NTT可以取模，且精度误差小。

我们的唯一限制就是取模的质数p=k*2^n+1，因此998244353应运而生。

对于如何构造使得每次变换都会减少一半的长度这个问题和p的原根有关，在这里就不讲了。

然而对于p不确定的时候，我们也可以使用中国剩余定理。

具体来说，找到一些p1，p2……pk满足NTT条件，然后计算结果，最后用中国剩余定理依次消即可。

然而这题很恶心的是很有可能爆longlong，且在模数大于int的情况下也没法快速乘，这时候就要使用骆克强提出的快速乘了（具体可以前往参考处第一篇博客。）

#include<cstdio>
#include<cctype>
#include<cstring>
#include<cmath>
#include<algorithm>
#include<iostream>
using namespace std;
typedef long long ll;
typedef long double dl;
const int N=5e5+;
const ll p1=,p2=,p3=,g=;
const ll M=p1*p2;
inline int read(){
    int X=,w=;char ch=;
    while(!isdigit(ch)){w|=ch=='-';ch=getchar();}
    while(isdigit(ch))X=(X<<)+(X<<)+(ch^),ch=getchar();
    return w?-X:X;
}
ll qpow(ll a,ll n,ll p){
    ll res=;
    while(n){
    if(n&)res=res*a%p;
    a=a*a%p;n>>=;
    }
    return res;
}
ll qmulti(ll a,ll b,ll p){
    a%=p,b%=p;
    return ((a*b-(ll)((ll)((dl)a/p*b+0.5)*p))%p+p)%p;
}
void FNT(ll a[],int n,int on,ll p){
    for(int i=,j=n>>;i<n-;i++){
        if(i<j)swap(a[i],a[j]);
        int k=n>>;
        while(j>=k){j-=k;k>>=;}
        if(j<k)j+=k;
    }
    for(int i=;i<=n;i<<=){
    ll res=qpow(g,(p-)/i,p);
        for(int j=;j<n;j+=i){
        ll w=;
            for(int k=j;k<j+i/;k++){
                ll u=a[k],t=w*a[k+i/]%p;
                a[k]=(u+t)%p;
                a[k+i/]=(u-t+p)%p;
                w=w*res%p;
            }
        }
    }
    if(on==-){
    ll inv=qpow(n,p-,p);
    a[]=a[]*inv%p;
    for(int i=;i<=n/;i++){
        a[i]=a[i]*inv%p;
        if(i!=n-i)a[n-i]=a[n-i]*inv%p;
        swap(a[i],a[n-i]);
    }
    }
}
int n,m,p;
ll a[N],b[N],c[N],d[N],ans[][N];
int main(){
    n=read(),m=read(),p=read();
    for(int i=;i<=n;i++)a[i]=read();
    for(int i=;i<=m;i++)b[i]=read();
    int nn=;
    while(nn<=n+m)nn<<=;

    memcpy(c,a,sizeof(a));memcpy(d,b,sizeof(b));
    FNT(c,nn,,p1);FNT(d,nn,,p1);
    for(int i=;i<nn;i++)ans[][i]=c[i]*d[i]%p1;
    memset(c,,sizeof(c));memset(d,,sizeof(d));

    memcpy(c,a,sizeof(a));memcpy(d,b,sizeof(b));
    FNT(c,nn,,p2);FNT(d,nn,,p2);
    for(int i=;i<nn;i++)ans[][i]=c[i]*d[i]%p2;
    memset(c,,sizeof(c));memset(d,,sizeof(d));

    memcpy(c,a,sizeof(a));memcpy(d,b,sizeof(b));
    FNT(c,nn,,p3);FNT(d,nn,,p3);
    for(int i=;i<nn;i++)ans[][i]=c[i]*d[i]%p3;
    memset(c,,sizeof(c));memset(d,,sizeof(d));

    FNT(ans[],nn,-,p1);
    FNT(ans[],nn,-,p2);
    FNT(ans[],nn,-,p3);

    for(int i=;i<=n+m;i++){
    ll A=(qmulti(ans[][i]*p2%M,qpow(p2%p1,p1-,p1),M)+
          qmulti(ans[][i]*p1%M,qpow(p1%p2,p2-,p2),M))%M;
    ll k=((ans[][i]-A)%p3+p3)%p3*qpow(M%p3,p3-,p3)%p3;
    printf("%lld ",((k%p)*(M%p)%p+A%p)%p);
    }
    puts("");
    return ;
}

+++++++++++++++++++++++++++++++++++++++++++

+本文作者：luyouqi233。 　　　　　　　　　　　　　　+

+欢迎访问我的博客：http://www.cnblogs.com/luyouqi233/+

+++++++++++++++++++++++++++++++++++++++++++