两个多项式的卷积(对询问分块)

两个多项式的卷积

思路：对询问分块，当凑够一定数量的修改时进行一次NTT，然后每次查询就直接暴力求还没有进行NTT的那一部分的贡献，块大小为sqrt(n*log(n))最合适，注意坑点：输入的a[i]和b[i]是有可能小于0的！

#include<iostream>
#include<cstdio>
#include<cmath>
#include<algorithm>
using namespace std;
const int MAX_N=100100;
const long long MOD=998244353;
const long long G=3;
const long long Gi=332748118;
long long a[MAX_N],b[MAX_N];
int lim;
int st_p[MAX_N],st_q[MAX_N],cnt=0;
long long A[2*MAX_N],B[2*MAX_N];
long long Sum[2*MAX_N],sum[MAX_N];
long long ask(int l,int r){
    if(r<0||l>r)
    return 0;
    l=max(l,0);
    if(l!=0)
    return (sum[r]-sum[l-1]+MOD)%MOD;
    else
    return sum[r];
}
long long Ask(int l,int r){
    if(r<0)
    return 0;
    l=max(l,0);
    if(l!=0)
    return (Sum[r]-Sum[l-1]+MOD)%MOD;
    else
    return Sum[r];
}
int limit=1,L,r[MAX_N];
inline long long pow_mod(long long a,long long n,long long m){
    long long ans=1;
    while(n){
        if(n&1){
            ans=(ans*a)%m;
        }
        a=(a*a)%m;
        n>>=1;
    }
    return ans;
}
inline void NTT(long long *A,int type) {
    for(int i=0;i<limit;i++){
        if(i<r[i])
        swap(A[i],A[r[i]]);
    }
    for(int mid=1;mid<limit;mid<<=1){    
        long long Wn=pow_mod(type==1?G:Gi,(MOD-1)/(mid<<1),MOD);
        for(int j=0;j<limit;j+=(mid<<1)){
            long long w=1;
            for(int k=0;k<mid;k++,w=(w*Wn)%MOD){
                int x=A[j+k],y=w*A[j+k+mid]%MOD;
                A[j+k]=(x+y)%MOD,
                A[j+k+mid]=(x-y+MOD)%MOD;
            }
        }
    }
}
void solve(int n,int m){//a[i]即为多项式乘法后x^i对MOD取模后的系数 
    limit=1;
    L=0;
    while(limit<=n+m){
        limit<<=1,L++;
    }
    for(int i=0;i<limit;i++)
    r[i]=(r[i>>1]>>1)|((i&1)<<(L-1));
    for(int i=n+1;i<limit;i++)
    A[i]=0;
    for(int i=m+1;i<limit;i++)
    B[i]=0;
    NTT(A,1);
    NTT(B,1);    
    for(int i=0;i<limit;i++)
    A[i]=(A[i]*B[i])%MOD;
    NTT(A,-1);    
    long long inv=pow_mod(limit,MOD-2,MOD);
    for(int i=0;i<=n+m;i++)
    A[i]=A[i]*inv%MOD;
}
int n;
void build(){
    int i;
    for(i=1;i<=cnt;i++)
    a[st_p[i]]=(a[st_p[i]]+st_q[i]+MOD)%MOD; 
    for(i=0;i<=n;i++)
    A[i]=a[i],B[i]=b[i];
    solve(n,n);
    Sum[0]=A[0];
    for(i=1;i<=2*n;i++)
    Sum[i]=(Sum[i-1]+A[i]+MOD)%MOD;
}
int main(void){
    int i,j,m,op,x,y;
    scanf("%d",&n);
    for(i=0;i<=n;i++){
        scanf("%lld",&a[i]);
        a[i]=(a[i]+MOD)%MOD;
    }
    for(i=0;i<=n;i++){
        scanf("%lld",&b[i]);
        b[i]=(b[i]+MOD)%MOD;
    }
    scanf("%d",&m);
    lim=sqrt(m*log2(n));
    sum[0]=b[0];
    for(i=1;i<=n;i++)
    sum[i]=(sum[i-1]+b[i])%MOD;
    build();
    //cout<<lim<<"\n";
    for(i=1;i<=m;i++){
        scanf("%d%d%d",&op,&x,&y);
        if(op==1){
            long long ans=0;
            ans=(ans+Ask(x,y))%MOD;
            for(j=1;j<=cnt;j++){
                int p=st_p[j];
                long long q=st_q[j];
                //影响了[p,p+n]区间
                int l=x,r=y;
                if(l>p+n||r<p)
                continue;
                l=max(l,p);
                r=min(r,p+n);
                ans=(ans+(q+MOD)%MOD*ask(l-p,r-p)%MOD)%MOD;
            }
            printf("%lld\n",ans);
        }
        else if(op==2){
            st_p[++cnt]=x;
            st_q[cnt]=y;
        }
        if(cnt==lim){
            build();
            cnt=0;
        }
    }
    return 0;
}