[bzoj4836]二元运算

[bzoj4836]二元运算

---

分治FFT

这题卡常啊有毒。

目前dark 3s bzoj T

• 代码

#include<bits/stdc++.h>
using namespace std;
typedef double db;
typedef complex<db> cplx;
const db PI=acos(-1.00);
const int N=3e5+5;
const int top=1<<19;
int rev[N];
inline void DFT(cplx *t,int n,int type){
    for(int i=0;i<n;i++)
        if(i<rev[i])swap(t[i],t[rev[i]]);
    for(int step=1;step<n;step<<=1){
        cplx wn=cplx(cos(PI/step),type*sin(PI/step));//主单位根 
        for(int T=step<<1,j=0;j<n;j+=T){
            cplx w=cplx(1.0,0);
            for(int k=0;k<step;k++,w*=wn){
                cplx x=t[j+k],y=t[j+k+step]*w;
                t[j+k]=x+y,t[j+k+step]=x-y;
            }
        }
    }
    if(type==-1) for(int i=0;i<n;i++) t[i]/=(db)n;
}
cplx A[N],B[N],C[N],D[N];
cplx tmpA[N],tmpB[N],tmpC[N];
void FFT(int bid,int aid,int toid,cplx *tar,int n){
    for(int i=0;i<n;i++) tmpB[i]=A[aid+i],tmpA[i]=B[bid+i];
    for(int i=n;i<n*2;i++)tmpA[i]=tmpB[i]=cplx(0.00,0.00);
    int l=0;while(1<<l < n*2)++l;
    for(int i=0;i<n*2;i++)rev[i]=(rev[i>>1]>>1)|((i&1)<<(l-1));
    DFT(tmpA,n*2,1);
    DFT(tmpB,n*2,1);
    for(int i=0;i<n*2;i++)tmpC[i]=tmpA[i]*tmpB[i];
    DFT(tmpC,n*2,-1);
    for(int i=0;i<n*2;i++)tar[toid+i]+=tmpC[i];
}
int mxA,mxB,mx;
inline void solve(int l,int r){
    if(l==r)return;
    int mid=(l+r)>>1;
    int len=mid-l+1;
    if(mid+1<=mxB) FFT(mid+1,l,l+mid+1,C,len);
    solve(l,mid);
    if(mid+1<=mxB)solve(mid+1,r);
}
 
int main()
{
    int T;scanf("%d",&T);
    while(T--){
        int n,m,q;scanf("%d%d%d",&n,&m,&q);
        mxA=0,mxB=0,mx=0;
        memset(A,0,sizeof(A)),memset(B,0,sizeof(B));
        memset(C,0,sizeof(C)),memset(D,0,sizeof(D));
        for(int i=1;i<=n;i++){int s;scanf("%d",&s);A[s]+=1.00;mxA=max(mxA,s);}
        for(int i=1;i<=m;i++){int s;scanf("%d",&s);B[s]+=1.00;mxB=max(mxB,s);}
        int top=1;mx=max(mxA+2,mxB+2);
        while(top<=mx*2)top<<=1;
        solve(0,top-1);
        for(int i=0;i<top/2;i++)if(top/2-i-1>i)swap(B[top/2-i-1],B[i]);
        FFT(0,0,0,D,top/2);
        while(q--){
            int id;scanf("%d",&id);
            printf("%.0lf\n",fabs(round(C[id].real()+D[id+top/2-1].real())));
        }
    }
}