bzoj 4836 二元运算 - 分治FFT

傻逼分治FFT题。

#include<iostream>
#include<cstring>
#include<cstdio>
#include<algorithm>
#include<cmath>
#include<assert.h>
#define N 400010
#define gc getchar()
#define lint long long
#define db double
#define Pi acos(-1)
#define debug(x) cerr<<#x<<"="<<x
#define sp <<" "
#define ln <<endl
using namespace std;
inline int inn()
{
    int x,ch;while((ch=gc)<'0'||ch>'9');
    x=ch^'0';while((ch=gc)>='0'&&ch<='9')
        x=(x<<1)+(x<<3)+(ch^'0');return x;
}
struct E{
    db x,y;
    E(db _x=0.0,db _y=0.0) { x=_x,y=_y; }
    inline E operator=(const E &w)
    {   return x=w.x,y=w.y,*this;   }
    inline E operator+(const E &w)
    {   return E(x+w.x,y+w.y);  }
    inline E operator-(const E &w)
    {   return E(x-w.x,y-w.y);  }
    inline E operator*(const E &w)
    {   return E(x*w.x-y*w.y,x*w.y+y*w.x);  }
    inline E operator*=(const E &w)
    {   return (*this)=(*this)*w;   }
}a[N],b[N];lint ans[N];int r[N];
int c1[N],c2[N],c3[N],x[N],y[N];
inline int FFT(E *a,int n,int s)
{
    for(int i=1;i<n;i++) if(i<r[i]) swap(a[i],a[r[i]]);
    for(int i=1;i<n;i<<=1)
    {
        E wn(cos(Pi/i),s*sin(Pi/i));
        for(int j=0,t=i<<1;j<n;j+=t)
        {
            E w(1,0);
            for(int k=0;k<i;k++,w*=wn)
            {
                E x=a[j+k],y=w*a[j+k+i];
                a[j+k]=x+y,a[j+k+i]=x-y;
            }
        }
    }
    return 0;
}
inline int tms(int *A,int *B,lint *C,int m1,int m2)
{
    int n=1,L=0;while(n<=m1+m2) n<<=1,L++;
    for(int i=0;i<=m1;i++) a[i].x=A[i],a[i].y=0;
    for(int i=0;i<=m2;i++) b[i].x=B[i],b[i].y=0;
    for(int i=m1+1;i<n;i++) a[i].x=a[i].y=0;
    for(int i=m2+1;i<n;i++) b[i].x=b[i].y=0;
    for(int i=1;i<n;i++) r[i]=(r[i>>1]>>1)|((i&1)<<(L-1));
    FFT(a,n,1),FFT(b,n,1);
    for(int i=0;i<n;i++) a[i]*=b[i];
    FFT(a,n,-1);
    for(int i=0;i<=m1+m2;i++) C[i]+=(lint)(a[i].x/n+0.5);
    return 0;
}
int solve(int l,int r)
{
    if(l==r) return 0;int mid=(l+r)>>1;
    solve(l,mid),solve(mid+1,r);
    tms(c1+l,c2+mid+1,ans+l+mid+1,mid-l,r-mid-1);
    return 0;
}
int main()
{
    for(int T=inn();T;T--)
    {
        int n=inn(),m=inn(),q=inn(),mx=0,v;
        for(int i=1;i<=n;i++) mx=max(mx,x[i]=inn());
        for(int i=1;i<=m;i++) mx=max(mx,y[i]=inn());
        memset(c1,0,sizeof(int)*(mx+1));
        memset(c2,0,sizeof(int)*(mx+1));
        for(int i=1;i<=n;i++) c1[x[i]]++;
        for(int i=1;i<=m;i++) c2[y[i]]++;
        for(int i=0;i<=mx;i++) c3[i]=c2[mx-i];
        memset(ans,0,sizeof(lint)*(2*mx+1));
        tms(c1,c3,ans,mx,mx);
        for(int i=0;i<=mx;i++) ans[i]=ans[i+mx];
        for(int i=mx+1;i<=mx+mx;i++) ans[i]=0ll;
        solve(0,mx);
        while(q--) v=inn(),printf("%lld\n",(v<=mx*2)?ans[v]:0);
    }
    return 0;
}