NTT板子

博客内容涉及NTT(Number Theoretic Transform)在算法竞赛中的应用,作者关注NTT的速度和模数选择问题,最终决定使用UOJ的模数进行计算。

话说NTT真不是一般的快啊。。。。
还有到底用那个模数啊。。。。
UOJ模数会不会出问题啊。。。
算了,就用UOJ模数吧。。。
代码:

#include<cstdio>
#include<cstring>
#include<iostream>
#include<cmath>
#include<algorithm>
#include<cstdlib>
#define ll long long
#define max(a,b) a>b?a:b
#define min(a,b) a<b?a:b
using namespace std;
inline int read(){
    int x=0,f=1;char ch=' ';
    while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}
    while(ch>='0'&&ch<='9')x=(x<<3)+(x<<1)+(ch^48),ch=getchar();
    return x*f;
}
const int N=3e6,p=998244353,g=3,gi=332748118;
int n,m,L,R[N];
ll inv,wn[20],A[N],B[N];
inline ll ksm(ll a,ll n){
    ll ans=1;
    while(n){
        if(n&1)ans=ans*a%p;
        a=a*a%p;
        n>>=1;
    }
    return ans;
}
inline void NTT(ll *a,int f){
    for(int i=0;i<n;++i)if(i<R[i])swap(a[i],a[R[i]]);
    for(int i=1;i<n;i<<=1){
        ll wn=ksm((f==1)?g:gi,(p-1)/(i<<1));
        for(int j=0;j<n;j+=(i<<1)){
            ll w=1;
            for(int k=0;k<i;++k,w=w*wn%p){
                ll x=a[j+k],y=w*a[j+k+i]%p;
                a[j+k]=(x+y)%p;a[j+k+i]=(x-y+p)%p;
            }
        }
    }
    if(!~f)for(int i=0;i<n;++i)a[i]=a[i]*inv%p;
}
int main(){
    for(int i=0;i<20;++i)wn[i]=ksm(g,(p-1)/(1<<i));
    scanf("%d %d",&n,&m);
    for(int i=0;i<=n;++i)A[i]=read(),A[i]=(A[i]+p)%p;
    for(int i=0;i<=m;++i)B[i]=read(),B[i]=(B[i]+p)%p;
    m+=n;
    for(n=1;n<=m;n<<=1)++L;
    inv=ksm(n,p-2);
    for(int i=0;i<n;++i)R[i]=(R[i>>1]>>1)|((i&1)<<(L-1));
    NTT(A,1);NTT(B,1);
    for(int i=0;i<n;++i)A[i]=A[i]*B[i]%p;
    NTT(A,-1);
    for(int i=0;i<=m;++i)printf("%d ",A[i]);
    return 0;
}
评论 2
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值